+def int_to_suffix(n):
+ if n > 1000000 and n%1000000 == 0:
+ return str(n//1000000) + 'M'
+ elif n > 1000 and n%1000 == 0:
+ return str(n//1000) + 'K'
+ else:
+ return str(n)
+
+######################################################################
+
+if args.nb_train_samples%args.batch_size > 0 or args.nb_test_samples%args.batch_size > 0:
+ print('The number of samples must be a multiple of the batch size.')
+ raise
+
+if args.compress_vignettes:
+ VignetteSet = vignette_set.CompressedVignetteSet
+else:
+ VignetteSet = vignette_set.VignetteSet
+