Hugging Face's logo

Ievan
/
fb-torch-minigrid

Model card Files
xet
 Community
Ievan commited on
Commit
fcd0863
·
verified ·
1 Parent(s): b20e9f7

Upload fb-torch-minigrid_20260616_223442/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. fb-torch-minigrid_20260616_223442/config.json +123 -0
fb-torch-minigrid_20260616_223442/config.json ADDED
@@ -0,0 +1,123 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "TrainConfig",
3
+ "agent": {
4
+ "name": "FBFlowBCAgent",
5
+ "model": {
6
+ "name": "FBFlowBCModel",
7
+ "device": "cuda",
8
+ "archi": {
9
+ "name": "FBFlowBCModelArchiConfig",
10
+ "L_dim": 100,
11
+ "z_dim": 100,
12
+ "norm_z": true,
13
+ "f": {
14
+ "name": "ForwardArchi",
15
+ "hidden_dim": 1024,
16
+ "model": "simple",
17
+ "hidden_layers": 1,
18
+ "embedding_layers": 2,
19
+ "num_parallel": 2,
20
+ "ensemble_mode": "batch"
21
+ },
22
+ "b": {
23
+ "name": "BackwardArchi",
24
+ "hidden_dim": 256,
25
+ "hidden_layers": 2,
26
+ "norm": true
27
+ },
28
+ "actor": {
29
+ "name": "noise_conditioned_actor",
30
+ "model": "simple",
31
+ "hidden_dim": 1024,
32
+ "hidden_layers": 1,
33
+ "embedding_layers": 2
34
+ },
35
+ "left_encoder": {
36
+ "name": "Identity"
37
+ },
38
+ "rgb_encoder": {
39
+ "name": "drq",
40
+ "feature_dim": 256
41
+ },
42
+ "augmentator": {
43
+ "name": "Identity"
44
+ },
45
+ "actor_vf": {
46
+ "name": "SimpleVectorFieldArchiConfig",
47
+ "model": "simple",
48
+ "hidden_dim": 1024,
49
+ "hidden_layers": 1
50
+ }
51
+ },
52
+ "obs_normalizer": {
53
+ "name": "RGBNormalizerConfig"
54
+ },
55
+ "inference_batch_size": 500000,
56
+ "seq_length": 1,
57
+ "actor_std": 0.2,
58
+ "amp": false,
59
+ "actor_encode_obs": false
60
+ },
61
+ "train": {
62
+ "name": "FBFlowBCAgentTrainConfig",
63
+ "lr_f": 0.0001,
64
+ "lr_b": 0.0001,
65
+ "lr_actor": 0.0001,
66
+ "weight_decay": 0.0,
67
+ "clip_grad_norm": 0.0,
68
+ "f_target_tau": 0.01,
69
+ "b_target_tau": 0.01,
70
+ "ortho_coef": 1.0,
71
+ "train_goal_ratio": 0.5,
72
+ "fb_pessimism_penalty": 0.0,
73
+ "actor_pessimism_penalty": 0.0,
74
+ "stddev_clip": 0.3,
75
+ "q_loss_coef": 0.0,
76
+ "batch_size": 1024,
77
+ "discount": 0.99,
78
+ "use_mix_rollout": false,
79
+ "update_z_every_step": 150,
80
+ "z_buffer_size": 10000,
81
+ "bc_coeff": 0.0,
82
+ "flow_steps": 10,
83
+ "lr_actor_vf": 0.0003
84
+ },
85
+ "cudagraphs": false,
86
+ "compile": false
87
+ },
88
+ "env": {
89
+ "name": "minigrid",
90
+ "domain": "minigrid",
91
+ "task": "MiniGrid-Empty-8x8-v0",
92
+ "seed": 0,
93
+ "obs_type": "pixels",
94
+ "frame_stack": 3
95
+ },
96
+ "data": {
97
+ "name": "minigrid",
98
+ "domain": "minigrid",
99
+ "dataset_root": "datasets",
100
+ "load_n_episodes": 500,
101
+ "obs_type": "pixels",
102
+ "buffer_type": "parallel",
103
+ "future": 0.99,
104
+ "num_workers": 8,
105
+ "horizon": 1,
106
+ "batch_size_factor": 1
107
+ },
108
+ "relabel_dataset": false,
109
+ "work_dir": "/home/sultanan/fb/td_jepa/workdir/train_dmc/2026-6-16-13:7:9-GKSO",
110
+ "seed": 0,
111
+ "log_every_updates": 10000,
112
+ "num_train_steps": 3000000,
113
+ "checkpoint_every_steps": 250000,
114
+ "enable_comet": true,
115
+ "comet_mode": "online",
116
+ "comet_run_group": "experiments",
117
+ "project": "fb-torch-minigrid",
118
+ "hf_repo": "Ievan/fb-torch-minigrid",
119
+ "buffer_device": null,
120
+ "evaluations": [],
121
+ "eval_every_steps": 250000,
122
+ "tags": {}
123
+ }