Hugging Face's logo

ZichenYan
/
AION

Robotics
reinforcement-learning
drones
aerial-navigation
object-goal-navigation
Model card Files
xet
 Community
1
AION
File size: 1,481 Bytes
79a4647
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
{
    "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
}