fix main.train_cn

MetaAugment/autoaugment_learners/gru_learner.py

@@ -61,7 +61,7 @@ class gru_learner(aa_learner):
                 # GRU-specific attributes that aren't in all other aa_learners's
                 alpha=0.2,
                 cont_mb_size=4,
-                cont_lr=0.1):
+                cont_lr=0.03):
         """
         Args:
             alpha (float, optional): Exploration parameter. It is multiplied to 

MetaAugment/main.py

@@ -51,6 +51,7 @@ def train_child_network(child_network,
         device = torch.device('cpu')
     child_network = child_network.to(device=device)
     
+    total_val=torch.tensor([0.0])
     best_acc=torch.tensor([0.0])
     early_stop_cnt = 0