AION / ckpt /args.json
ZichenYan's picture
Upload folder using huggingface_hub
79a4647 verified
Raw
History Blame Contribute Delete
1.48 kB
{
"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
}