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

use parameters from mtgjamendo repo

parent 0d4fec2d
......@@ -8,10 +8,15 @@ def run():
logger.info(CURR_RUN_PATH)
exp = Experiment(save_dir=CURR_RUN_PATH)
# TODO fill their training parameters
# parameters used in the baseline (read from main.py and solver.py)
# n_epochs = 500
# lr = 1e-4
# num_class = 56
# batch_size = 32
if USE_GPU:
trainer = Trainer(gpus=[0], distributed_backend='ddp',
experiment=exp, max_nb_epochs=10, train_percent_check=1.0,
experiment=exp, max_nb_epochs=500, train_percent_check=1.0,
fast_dev_run=False)
else:
trainer = Trainer(experiment=exp, max_nb_epochs=1, train_percent_check=0.1,
......
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