| { | |
| "title": "baseline-18-4", | |
| "rollout_steps": 128, | |
| "seed": 1, | |
| "workers": 25, | |
| "max_episode_length": 50, | |
| "grid_size": 0.25, | |
| "success_distance": 1.5, | |
| "get_seen_data": true, | |
| "train_scenes": "[1-20]", | |
| "val_scenes": "[21-30]", | |
| "possible_targets": "FULL_OBJECT_CLASS_LIST", | |
| "train_targets": null, | |
| "fov": 90.0, | |
| "vis": false, | |
| "snapToGrid": false, | |
| "partial_reward": true, | |
| "room_results": false, | |
| "episode_type": "NavTrainEpisode", | |
| "scene_types": [ | |
| "kitchen", | |
| "living_room", | |
| "bedroom", | |
| "bathroom" | |
| ], | |
| "lr": 0.0001, | |
| "gamma": 0.99, | |
| "tau": 1.0, | |
| "beta": 0.01, | |
| "ep_save_freq": 50000.0, | |
| "optimizer": "SharedAdam", | |
| "save_model_dir": "trained_models", | |
| "log_dir": "runs/", | |
| "gpu_ids": [ | |
| 0, | |
| 1 | |
| ], | |
| "amsgrad": true, | |
| "train_thin": 250, | |
| "offline_data_dir": "./data/thor_v1_offline_data", | |
| "results_json": "metrics.json", | |
| "load_model": "", | |
| "model": "AIONg", | |
| "glove_dim": 300, | |
| "action_space": 6, | |
| "hidden_state_sz": 512, | |
| "compute_spl": false, | |
| "eval": false, | |
| "dropout_rate": 0.25, | |
| "add_clip_align": true, | |
| "add_stats": true, | |
| "add_rgb": true, | |
| "add_depth": true, | |
| "add_dis_reward": true, | |
| "add_bbox_reward": true, | |
| "add_parent_reward": true, | |
| "add_collision_reward": true, | |
| "split": "18/4", | |
| "scene": "ithor", | |
| "save_visuals": false, | |
| "save_episode_data": false | |
| } |