-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
126 lines (106 loc) · 2.4 KB
/
config.yaml
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
feature_net_kwargs:
rgb_feat:
cls: DummyImgFeat
output_dim: 512
image_model: gdino # Image models: mineclip, gdino, hf_gdino, image_encoder
compass:
cls: CompassMLP
input_dim: 4
hidden_dim: 128
output_dim: 128
hidden_depth: 2
gps:
cls: GPSMLP
hidden_dim: 128
output_dim: 128
hidden_depth: 2
# voxels:
# cls: FlattenedVoxelBlockEncoder
# embed_dim: 8
# hidden_dim: 128
# output_dim: 128
# hidden_depth: 2
# biome_id:
# cls: BiomeIDEmb
# embed_dim: 8
# prev_action:
# cls: PrevActionEmb
# embed_dim: 8
# prompt:
# cls: PromptEmbFeat
# output_dim: 512
feature_fusion:
output_dim: 512
hidden_depth: 0
actor:
hidden_dim: 256
hidden_depth: 3
critic:
hidden_dim: 256
hidden_depth: 3
mineclip:
arch: "vit_base_p16_fz.v2.t2"
hidden_dim: 512
image_feature_dim: 512
mlp_adapter_spec: "v0-2.t0"
pool_type: "attn.d2.nh8.glusw" # filled by variant
resolution: [160, 256]
ckpt:
path: /home/kenjic/documents/MinedojoPPO/weights/mineclip_weights/attn.pth # filled by users
checksum: "b5ece9198337cfd117a3bfbd921e56da"
groundingdino:
ckpt: /home/kenjic/documents/MinedojoPPO/weights/gdino_weights/groundingdino_swint_ogc.pth # filled by user
setfile: /home/kenjic/documents/MinedojoPPO/GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py # filled by user
env:
task: "combat spider"
num_envs: 2
agent:
total_timesteps: 800000
num_steps: 10000
num_minibatches: 4
learning_epochs: 16
clip_vloss: true
return_norm: true
train_image_model: false
load_ppo_model: false
ppo_checkpoint_path: ""
load_image_model: false
image_checkpoint_path: ""
autocast_flag: false
gamma: 0.99
gae_lambda: 0.95
seed: 1
clip_coef: 0.2
learning_rate: !!float 1e-4 #1e-4
min_lr: !!float 1e-6
ent_coef: 0.005
vf_coef: 0.5
max_grad_norm: 10
target_kl: 0.015
debug: false
ppo_buffer:
buffer_size: 15000
number_actions: 89
si_buffer:
buffer_size: 50
buffer_mean: 0.1
buffer_std: 0
buffer_delta: 0
hyperparameters:
task: "combat spider"
epochs: 20
batch_size: 5000
video_min_rew: 32
load_checkpoint: false
checkpoint_dir: "/home/user/Mine/results/run_231109_0814/weights"
checkpoint_epoch: 0
imitation_learning: false
demos_path: ""
file_logging: true
wandb_init: true
# hydra:
# job:
# chdir: true
# run:
# dir: .
# output_subdir: null