parameters

pengpaiSH · Dec 2, 2016 · 0c6c9b2 · 0c6c9b2
1 parent 832c1d9
commit 0c6c9b2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -11,7 +11,7 @@
 img_height = 299
 nbr_train_samples = 3019
 nbr_validation_samples = 758
-nbr_epochs = 50
+nbr_epochs = 25
 batch_size = 32
 
 train_data_dir = '/Users/pengpai/Desktop/python/DeepLearning/Kaggle/NCFM/data/train_split'
@@ -60,6 +60,8 @@
         target_size = (img_width, img_height),
         batch_size = batch_size,
         shuffle = True,
+        # save_to_dir = '/Users/pengpai/Desktop/python/DeepLearning/Kaggle/NCFM/data/visualization',
+        # save_prefix = 'aug',
         classes = FishNames,
         class_mode = 'categorical')
 
@@ -68,6 +70,8 @@
         target_size=(img_width, img_height),
         batch_size=batch_size,
         shuffle = True,
+        #save_to_dir = '/Users/pengpai/Desktop/python/DeepLearning/Kaggle/NCFM/data/visulization',
+        #save_prefix = 'aug',
         classes = FishNames,
         class_mode = 'categorical')