diff --git a/MetaAugment/__pycache__/main.cpython-38.pyc b/MetaAugment/__pycache__/main.cpython-38.pyc deleted file mode 100644 index 5dcce355ebb82a6ff165d8bf473700ee3f54eae9..0000000000000000000000000000000000000000 Binary files a/MetaAugment/__pycache__/main.cpython-38.pyc and /dev/null differ diff --git a/MetaAugment/autoaugment_learners/gru_learner.py b/MetaAugment/autoaugment_learners/gru_learner.py index e23178e917b47a8c3f5e477603b3c9eb4673808e..bd305ff20693cfb54c4f2c2006fa08ff49e2e9f3 100644 --- a/MetaAugment/autoaugment_learners/gru_learner.py +++ b/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) diff --git a/check_pickles.py b/check_pickles.py new file mode 100644 index 0000000000000000000000000000000000000000..88f57da763684f3894971d7f7d1c472a183b974a --- /dev/null +++ b/check_pickles.py @@ -0,0 +1,8 @@ +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