add , in init
This commit is contained in:
parent
3374ad7620
commit
e10a178010
|
@ -25,7 +25,8 @@ class PSELoss(nn.Layer):
|
||||||
ohem_ratio=3,
|
ohem_ratio=3,
|
||||||
kernel_sample_mask='pred',
|
kernel_sample_mask='pred',
|
||||||
reduction='sum',
|
reduction='sum',
|
||||||
eps=1e-6**kwargs):
|
eps=1e-6,
|
||||||
|
**kwargs):
|
||||||
"""Implement PSE Loss.
|
"""Implement PSE Loss.
|
||||||
"""
|
"""
|
||||||
super(PSELoss, self).__init__()
|
super(PSELoss, self).__init__()
|
||||||
|
|
Loading…
Reference in New Issue