Load policy config in evaluation

2022-02-28 12:23:02 +01:00
parent 62c28d0cfa
commit a37995694d
3 changed files with 33 additions and 9 deletions
--- a/shail-experiment.py
+++ b/shail-experiment.py
@@ -247,6 +247,13 @@ if __name__ == '__main__':
            os.makedirs(savepath)
        
        import shutil
+
+        # save config
+        shutil.copyfile(
+            args.test,
+            os.path.join(savepath, 'config.json')
+        )
+
        for i in range(args.test_seeds):
            s = analysis._checkpoints[i]['config']['seed']
            check_dir = analysis._checkpoints[i]['logdir']