Skip to content
Snippets Groups Projects
Unverified Commit b9a81b09 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

Add gradient clip back in cfgs (#2608)


* Add gradient clip back in cfgs

* Reformat cfg

Co-authored-by: default avatarzww <zhangwenwei@sensetime.com>
parent bb67350a
No related branches found
No related tags found
No related merge requests found
......@@ -6,3 +6,5 @@ model = dict(
# learning policy
lr_config = dict(step=[16, 22])
total_epochs = 24
optimizer_config = dict(
_delete_=True, grad_clip=dict(max_norm=35, norm_type=2))
......@@ -18,3 +18,5 @@ model = dict(
target_means=[.0, .0, .0, .0],
target_stds=[0.1, 0.1, 0.2, 0.2]),
loss_bbox=dict(type='SmoothL1Loss', beta=0.11, loss_weight=0.75)))
optimizer_config = dict(
_delete_=True, grad_clip=dict(max_norm=35, norm_type=2))
......@@ -15,3 +15,5 @@ model = dict(
bins=10,
momentum=0.7,
loss_weight=10.0)))
optimizer_config = dict(
_delete_=True, grad_clip=dict(max_norm=35, norm_type=2))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment