"official/vision/ops/augment_test.py" did not exist on "999fae620807dc9c7425ae814a1822836335d427"
schedule_3x.py 343 Bytes
Newer Older
liyinhao's avatar
liyinhao committed
1
# optimizer
zhangwenwei's avatar
zhangwenwei committed
2
3
# This schedule is mainly used by models on indoor dataset,
# e.g., VoteNet on SUNRGBD and ScanNet
liyinhao's avatar
liyinhao committed
4
5
6
7
8
9
lr = 0.008  # max learning rate
optimizer = dict(type='Adam', lr=lr)
optimizer_config = dict(grad_clip=dict(max_norm=10, norm_type=2))
lr_config = dict(policy='step', warmup=None, step=[24, 32])
# runtime settings
total_epochs = 36