-
Notifications
You must be signed in to change notification settings - Fork 6
/
Point-OAE_2k.yaml
47 lines (42 loc) · 990 Bytes
/
Point-OAE_2k.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
optimizer : {
type: AdamW,
kwargs: {
lr : 0.0005,
weight_decay : 0.05
}}
scheduler: {
type: CosLR,
kwargs: {
epochs: 300,
initial_epochs : 3
}}
dataset : {
train : { _base_: cfgs/dataset_configs/ShapeNet-55.yaml,
others: {subset: 'train', npoints: 2048, whole: True}},
val : { _base_: cfgs/dataset_configs/ModelNet40.yaml,
others: {subset: 'test', bs: 4}},
extra_train : { _base_: cfgs/dataset_configs/ModelNet40.yaml,
others: {subset: 'train', bs: 4}}}
model : {
NAME: Point_OAE,
mask_rand: True,
use_cutmix: False,
use_emd: False,
transformer_config: {
mask_ratio: [0.75, 0.75],
trans_dim: 384,
depth_encoder: 12,
depth_decoder: 12,
drop_path_rate: 0.1,
cls_dim: 512,
num_heads: 6,
group_size: 32,
num_group: 128,
encoder_dims: 256,
}}
total_bs : 128 #64
step_per_update : 1
max_epoch : 300
clip_gradients: True
clip_grad: 3.0
consider_metric: CDL1