add , in init

This commit is contained in:
WenmuZhou 2021-09-01 17:04:18 +08:00
parent 3374ad7620
commit e10a178010
1 changed files with 2 additions and 1 deletions

View File

@ -25,7 +25,8 @@ class PSELoss(nn.Layer):
ohem_ratio=3,
kernel_sample_mask='pred',
reduction='sum',
eps=1e-6**kwargs):
eps=1e-6,
**kwargs):
"""Implement PSE Loss.
"""
super(PSELoss, self).__init__()