From 31fa29586377ce48bfe05082c1da34c9761f800f Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Fri, 8 Apr 2022 10:46:47 +0100
Subject: [PATCH] gru_learner also saves itself as pkl
---
MetaAugment/autoaugment_learners/gru_learner.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/MetaAugment/autoaugment_learners/gru_learner.py b/MetaAugment/autoaugment_learners/gru_learner.py
index 377064a2..e23178e9 100644
--- a/MetaAugment/autoaugment_learners/gru_learner.py
+++ b/MetaAugment/autoaugment_learners/gru_learner.py
@@ -156,10 +156,11 @@ class gru_learner(aa_learner):
cont_optim.step()
# save the history every 1 epochs as a pickle
- if _%1==1:
- with open('gru_logs.pkl', 'wb') as file:
- pickle.dump(self.history, file)
-
+ with open('gru_logs.pkl', 'wb') as file:
+ pickle.dump(self.history, file)
+ with open('gru_learner.pkl', 'wb') as file:
+ pickle.dump(self, file)
+
@@ -182,4 +183,4 @@ if __name__=='__main__':
learner = gru_learner(discrete_p_m=False)
newpol = learner.generate_new_policy()
learner.learn(train_dataset, test_dataset, child_network, toy_flag=True)
- pprint(learner.history)
\ No newline at end of file
+ pprint(learner.history)
--
GitLab