-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.yaml
127 lines (117 loc) · 2.91 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
127
data:
root_dir: ''
num_points: 16384
num_object_points: 1024
world_coord: True
num_rotations: 8
grid_resolution: 0.01
jitter_scale: 0.
contact_radius: 0.005
robot_prob: 1.0
offset_bins: [
0, 0.00794435329, 0.0158887021, 0.0238330509,
0.0317773996, 0.0397217484, 0.0476660972,
0.055610446, 0.0635547948, 0.0714991435, 0.08
]
m2t2:
scene_encoder:
type: 'pointnet2_msg'
num_points: 16384
downsample: 4
radius: 0.05
radius_mult: 2
use_rgb: False
object_encoder:
type: 'pointnet2_msg_cls'
num_points: 1024
downsample: 4
radius: 0.05
radius_mult: 2
use_rgb: False
contact_decoder:
mask_feature: 'res0'
in_features: ['res1', 'res2', 'res3']
place_feature: 'res4'
object_in_features: ['res1', 'res2', 'res3']
embed_dim: 256
feedforward_dim: 512
num_scales: 3
num_layers: 9
num_heads: 8
num_grasp_queries: 100
num_place_queries: 8
language_context_length: 0
language_token_dim: 256
use_attn_mask: True
use_task_embed: True
activation: 'GELU'
action_decoder:
use_embed: False
max_num_pred: null
hidden_dim: 256
num_layers: 2
num_params: 0
activation: 'GELU'
offset_bins: [
0, 0.00794435329, 0.0158887021, 0.0238330509,
0.0317773996, 0.0397217484, 0.0476660972,
0.055610446, 0.0635547948, 0.0714991435, 0.08
]
matcher:
object_weight: 2.0
bce_weight: 5.0
dice_weight: 5.0
grasp_loss:
object_weight: 2.0
not_object_weight: 0.1
pseudo_ce_weight: 0.0
bce_topk: 512
bce_weight: 5.0
dice_weight: 5.0
deep_supervision: True
recompute_indices: True
adds_pred2gt: 100.0
adds_gt2pred: 0.0
adds_per_obj: False
contact_dir: 0.0
approach_dir: 0.0
offset: 1.0
param: 1.0
offset_bin_weights: [
0.16652107, 0.21488856, 0.37031708, 0.55618503, 0.75124664,
0.93943357, 1.07824539, 1.19423112, 1.55731375, 3.17161779
]
place_loss:
bce_topk: 1024
bce_weight: 5.0
dice_weight: 5.0
deep_supervision: True
optimizer:
type: 'ADAMW'
base_batch_size: 16
base_lr: 0.0001
backbone_multiplier: 1.0
grad_clip: 0.01
weight_decay: 0.05
train:
mask_thresh: 0.5
num_gpus: 8
port: '1234'
batch_size: 16
num_workers: 8
num_epochs: 160
print_freq: 25
plot_freq: 50
save_freq: 10
checkpoint: null
log_dir: ''
eval:
data_dir: ''
checkpoint: ''
mask_thresh: 0.4
object_thresh: 0.4
num_runs: 1
world_coord: True
surface_range: 0.02
placement_height: 0.02
placement_vis_radius: 0.3