Skip to content

Commit de8612a

Browse files
committed
Undo mistake in previous commit
1 parent 09ff37d commit de8612a

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

Pilot1/Uno/uno_baseline_keras2.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,6 @@ def warmup_scheduler(epoch):
375375

376376
ckpt.set_model(template_model)
377377
J = ckpt.restart(params)
378-
# J = candle.restart(params, model)
379378

380379
if J is not None:
381380
initial_epoch = J['epoch']
@@ -384,10 +383,6 @@ def warmup_scheduler(epoch):
384383
print('initial_epoch: %i' % initial_epoch)
385384

386385
elif args.initial_weights is not None:
387-
# ckpt = candle.CandleCheckpointCallback(params,
388-
# verbose=True)
389-
390-
# if args.initial_weights:
391386
logger.info("Loading initial weights from '{}'"
392387
.format(args.initial_weights))
393388
start = time.time()
@@ -425,7 +420,7 @@ def warmup_scheduler(epoch):
425420
es_monitor = keras.callbacks.EarlyStopping(patience=patience,
426421
verbose=1)
427422

428-
reduce_lr = ReduceLROnPlateau(monitor='val_loss', factor=0.5, patience=100, min_lr=0.00001)
423+
reduce_lr = ReduceLROnPlateau(monitor='val_loss', factor=0.5, patience=5, min_lr=0.00001)
429424
warmup_lr = LearningRateScheduler(warmup_scheduler)
430425
# prefix + cv_ext + '.
431426
checkpointer = MultiGPUCheckpoint('model.h5', save_best_only=True)

0 commit comments

Comments
 (0)