Dont save the model

ssbpv · Jul 27, 2018 · 75b6341 · 75b6341
1 parent 8ccedd9
commit 75b6341
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras-color/color.py b/keras-color/color.py
@@ -69,5 +69,5 @@ def my_generator(batch_size, img_dir):
 
 model.fit_generator( my_generator(config.batch_size, train_dir),
                      samples_per_epoch=20,
-                     nb_epoch=config.num_epochs, callbacks=[WandbCallback(data_type='image')],
+                     nb_epoch=config.num_epochs, callbacks=[WandbCallback(data_type='image', save_model=False)],
                      validation_data=(val_bw_images, val_color_images))