increase max batch for toy where val is needed

confs/toy.yaml

@@ -20,7 +20,7 @@ nas:
         train_batch: 32
         test_batch: 64
         dataset:
-          max_batches: 2
+          max_batches: 4
     post_train:
       trainer:
         epochs: 1
@@ -28,7 +28,7 @@ nas:
         train_batch: 32
         test_batch: 64
         dataset:
-          max_batches: 2
+          max_batches: 4
     model_desc:
       n_reductions: 1 # number of reductions to be applied
       n_cells: 3 # number of cells
@@ -37,7 +37,7 @@ nas:
       train_batch: 32
       test_batch: 64
       dataset:
-        max_batches: 2
+        max_batches: 4
     trainer:
       epochs: 1
   eval: