Commit deeb26d5 authored by Verena Praher's avatar Verena Praher
Browse files

change resnet parameters to speed up training for now

parent 2f72c41d
......@@ -15,7 +15,7 @@ def run():
# callbacks
early_stop = EarlyStopping(
monitor='val_loss',
patience=50,
patience=20,
verbose=True,
mode='min'
)
......@@ -30,7 +30,7 @@ def run():
if USE_GPU:
trainer = Trainer(gpus=[0], distributed_backend='ddp',
experiment=exp, max_nb_epochs=500, train_percent_check=1.0,
experiment=exp, max_nb_epochs=100, train_percent_check=1.0,
fast_dev_run=False, early_stop_callback=early_stop,
checkpoint_callback=checkpoint_callback)
else:
......@@ -42,7 +42,6 @@ def run():
print(model)
trainer.fit(model)
# TODO log testing results
trainer.test()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment