Fix BUG: 'TypeError: <lambda>() takes 1 positional argument but 2 wer…

…e given'
cl0ver012 · May 11, 2018 · 8b28ec0 · 8b28ec0
1 parent 0e25a77
commit 8b28ec0
Show file tree

Hide file tree

Showing 5 changed files with 42,949 additions and 42,936 deletions.
diff --git a/train.py b/train.py
@@ -29,7 +29,7 @@
     model_checkpoint = ModelCheckpoint(model_names, monitor='val_loss', verbose=1, save_best_only=True)
     early_stop = EarlyStopping('val_loss', patience=patience)
     reduce_lr = ReduceLROnPlateau('val_loss', factor=0.1, patience=int(patience / 4), verbose=1)
-    cleanup = LambdaCallback(on_epoch_begin=lambda epoch, logs: trimap_clear())
+    cleanup = LambdaCallback(on_epoch_begin=lambda epoch, logs: trimap_clear(epoch))
     callbacks = [tensor_board, model_checkpoint, early_stop, reduce_lr, cleanup]
 
     # Start Fine-tuning