Update config.yaml

Author: licksylick

classification_models_pytorch/config.yaml

@@ -16,7 +16,6 @@ trainer:
   target: pytorch_lightning.Trainer
   params:
 #    gpus: [0]
-    max_epochs: 20
 
 model:
   target: models.ClassificationModel
@@ -55,4 +54,4 @@ callbacks:
   params:
     monitor: val_loss
     patience: 5
-    mode: min
+    mode: min