Make no of samples multiple of batch size

transfer-learning.py

@@ -505,8 +505,11 @@ datagen = ImageDataGenerator(
     vertical_flip=False
 )
 
+val_samples = int(x_train.shape[0] * 0.1)
+# round up to nearest multiple of batch_size
+val_samples += batch_size - (val_samples % batch_size)
 x_train, x_val, y_train, y_val = train_test_split(
-    x_train, y_train, test_size=0.1) 
+    x_train, y_train, test_size=val_samples)
 datagen.fit(x_train)
 
 convs = 1