Skip to content

Commit 831ba64

Browse files
authored
[Fix] Fix WIDERFace SSD loss NAN. (#9734)
1 parent e7cedcd commit 831ba64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

configs/wider_face/ssd300_wider_face.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
model = dict(bbox_head=dict(num_classes=1))
66
# optimizer
77
optimizer = dict(type='SGD', lr=0.012, momentum=0.9, weight_decay=5e-4)
8-
optimizer_config = dict()
8+
optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))
99
# learning policy
1010
lr_config = dict(
1111
policy='step',

0 commit comments

Comments
 (0)