add more configs to the baseline experiment

experiments/experiment_baseline.py

@@ -7,13 +7,28 @@ from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 import os
 
 config = {
-    'epochs':1
+    'epochs': 1,
+    'patience': 50,
+    'earlystopping_metric': 'val_loss', # 'val_prauc'
+    'earlystopping_mode': 'min' # 'max'
 }
 
 def epochs_500():
     global config
     config['epochs'] = 500
 
+
+def epochs_100():
+    global config
+    config['epochs'] = 100
+    config['patience'] = 20
+
+
+def epochs_20():
+    global config
+    config['epochs'] = 20
+
+
 def run(hparams):
     init_experiment(comment=hparams.experiment_name)
     from utils import CURR_RUN_PATH, logger # import these after init_experiment
@@ -48,10 +63,10 @@ def run(hparams):
     # batch_size = 32
 
     early_stop = EarlyStopping(
-        monitor='val_loss',
-        patience=50,
+        monitor=config['earlystopping_metric'],
+        patience=config['patience'],
         verbose=True,
-        mode='min'
+        mode=config['earlystopping_mode']
     )
 
     checkpoint_callback = ModelCheckpoint(