use parameters from mtgjamendo repo

experiments/experiment_baseline.py

@@ -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,