Skip to content

Commit fc05b6f

Browse files
committed
Better handling of best_metric_last (P1B1)
1 parent 8b969df commit fc05b6f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

Pilot1/P1B1/p1b1_baseline_pytorch.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -199,9 +199,7 @@ def fit(model, X_train, X_val, params):
199199
J = ckpt.restart(model)
200200
if J is not None:
201201
initial_epoch = J["epoch"]
202-
best_metric_last = J["best_metric_last"]
203-
params["ckpt_best_metric_last"] = best_metric_last
204-
print("initial_epoch: %i" % initial_epoch)
202+
print("restarting from ckpt: initial_epoch: %i" % initial_epoch)
205203

206204
total_step = len(train_iter)
207205
loss_list = []

0 commit comments

Comments
 (0)