undo fancy argparse changes

Author: staceysv

libs/training.py

@@ -14,11 +14,11 @@
 from wandb.fastai import WandbCallback
 
 
-def train_model(dataset, config):
+def train_model(dataset):
     """ Trains a DynamicUnet on the dataset """
 
-    epochs = config["epochs"]
-    lr     = config["lr"]
+    epochs = 15
+    lr     = 1e-4
     size   = 300
     wd     = 1e-2
     bs     = 8 # reduce this if you are running out of GPU memory