add , in init
This commit is contained in:
parent
3374ad7620
commit
e10a178010
|
@ -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__()
|
||||
|
|
Loading…
Reference in New Issue