Skip to content

Commit 8b969df

Browse files
committed
Better handling of best_metric_last
1 parent ae4abb3 commit 8b969df

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

Pilot1/NT3/nt3_baseline_keras2.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,6 @@ def run(gParameters):
8585
model = Sequential()
8686

8787
initial_epoch = 0
88-
best_metric_last = None
8988

9089
X_train, Y_train, X_test, Y_test = load_data(train_file, test_file, gParameters)
9190

@@ -172,9 +171,7 @@ def run(gParameters):
172171
J = ckpt.restart(model)
173172
if J is not None:
174173
initial_epoch = J["epoch"]
175-
best_metric_last = J["best_metric_last"]
176-
gParameters["ckpt_best_metric_last"] = best_metric_last
177-
print("initial_epoch: %i" % initial_epoch)
174+
print("restarting from ckpt: initial_epoch: %i" % initial_epoch)
178175

179176
# Reference case
180177
# model.add(Conv1D(filters=128, kernel_size=20, strides=1, padding='valid', input_shape=(P, 1)))

0 commit comments

Comments
 (0)