Commit e10a1780 authored by WenmuZhou's avatar WenmuZhou
Browse files

add , in init

parent 3374ad76
...@@ -25,7 +25,8 @@ class PSELoss(nn.Layer): ...@@ -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__()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment