tester update

myGym/configs/tester.json

@@ -19,7 +19,7 @@
 "used_objects":null,
 "object_sampling_area":[-0.7, 0.7, 0.2, 0.8, 0.1, 0.1],
 #Distractor
-"distractors" : ["sphere"],
+"distractors" : null,
 "distractor_moveable" : 1,
 "distractor_constant_speed" : 0,
 "distractor_movement_dimensions": 3,
@@ -39,13 +39,13 @@
 #Train
 "train_framework":"tensorflow",
 "algo":"ppo2",
-"max_episode_steps":512,
-"algo_steps":512,
-"steps":200000,
+"max_episode_steps":256,
+"algo_steps":256,
+"steps":20000,
 "pretrained_model":null,
 "multiprocessing":false,
 #Evaluation
-"eval_freq":50000,
+"eval_freq":21000,
 "eval_episodes":10,
 #Saving and logging
 "logdir":"trained_models/tester",

myGym/tester.sh

@@ -0,0 +1,5 @@
+#!/bin/sh
+rm -r ./trained_models/tester
+python train.py --config ./configs/tester.json
+python test.py -cfg ./trained_models/tester/reach_table_kuka_joints_gt_ppo2/train.json
+rm -r ./trained_models/tester

myGym/utils/visualize_results_srate.py

@@ -11,8 +11,8 @@
 
 def get_arguments():
  parser = argparse.ArgumentParser()
- parser.add_argument("-pth", default='/nfs/projects/crow/mygym/trained_models')
- parser.add_argument("-task", default='reach')
+ parser.add_argument("-pth", default='./trained_models/')
+ parser.add_argument("-task", default='press')
  parser.add_argument("-common", default='table_kuka_joints_gt')
  parser.add_argument("-algos", default=['ppo', 'ppo2', 'sac',], nargs='*')
  parser.add_argument("-xlabel", type=int, default=1)