From 2396b54bc088e10dc1b7ca1f428c6999be0d8a39 Mon Sep 17 00:00:00 2001
From: Ztira <aritz.martinez@tecnalia.com>
Date: Wed, 27 Jan 2021 10:04:15 +0100
Subject: [PATCH] Udate exp.py so it matches the rest of updates
---
exp.py | 17 ++++++++++++++++-
1 file changed, 16 insertions(+), 1 deletion(-)
diff --git a/exp.py b/exp.py
index 7e5a135..c6d23e2 100644
--- a/exp.py
+++ b/exp.py
@@ -10,6 +10,10 @@ from utils.callback import callback_function as cb
from utils.utils import load_compressed_pickle as lp
from pickle import dump as pdump
+PATH_TO_METAWORLD_FILE = './metaworld/envs/mujoco/sawyer_xyz/sawyer_xyz_env.py'
+PATH_TO_FIXED = './exp_type/sawyer_xyz_env_fixed.py'
+PATH_TO_RANDOM = './exp_type/sawyer_xyz_env_random.py'
+
def end_exp():
shutil.rmtree('/tmp/amfearl/', ignore_errors=True)
@@ -25,6 +29,9 @@ if __name__ == "__main__":
parser.add_argument('-p', dest='path', type=str, default="summary")
parser.add_argument('-t', dest='threads', type=int, default=None)
parser.add_argument('-exp', dest='experiment', type=int, default=0)
+ parser.add_argument('-r', dest='random', type=int, default=1,
+ help="0 - fixed ; 1 (default) - random")
+
args = parser.parse_args()
filename = args.path
@@ -39,6 +46,14 @@ if __name__ == "__main__":
envs = []
+ # Copy required files to make episodes' initialization random
+ if args.random:
+ shutil.copyfile(PATH_TO_RANDOM, PATH_TO_METAWORLD_FILE)
+ print("Experiment configured for random reinitialization")
+ else:
+ shutil.copyfile(PATH_TO_FIXED, PATH_TO_METAWORLD_FILE)
+ print("Experiment configured for fixed reinitialization")
+
if args.experiment == 0:
# TOY EXPERIMENT
print('Loading compressed_TOY.pbz2, this may take a few seconds...')
@@ -66,7 +81,7 @@ if __name__ == "__main__":
for i in range(len(tasks)):
pdump(tasks[i], open('/tmp/amfearl/'+str(i)+'.pickle', 'wb'))
- mfea.mfea(tasks, rmp=0.8, gen=1000, pop=10, episodes=10,
+ mfea.mfea(tasks, rmp=0.8, gen=1000, pop=60, episodes=300,
D_multitask=dim, callback=cb,
f_name=filename, num_threads=args.threads)
--
GitLab