Merge branch 'master' of gitlab.doc.ic.ac.uk:yw21218/metarl

MetaAugment/autoaugment_learners/gru_learner.py

@@ -52,7 +52,7 @@ class gru_learner(aa_learner):
 
         self.rnn_output_size = fun_num+p_bins+m_bins
         self.controller = RNNModel(mode='GRU', output_size=self.rnn_output_size, 
-                                    num_layers=1, bias=True)
+                                    num_layers=2, bias=True)
         self.softmax = torch.nn.Softmax(dim=0)
 
 

check_pickles.py

+import pickle
+from pprint import pprint
+
+with open('randomsearch_logs.pkl', 'rb') as file:
+    list = pickle.load(file)
+
+pprint(list)
+print(len(list))
\ No newline at end of file