Use pathlib in DOTA ops (#7552)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
f6309b8e70
commit
9d4ffa43bc
6 changed files with 19 additions and 55 deletions
|
|
@ -38,9 +38,7 @@ class VarifocalLoss(nn.Module):
|
|||
class FocalLoss(nn.Module):
|
||||
"""Wraps focal loss around existing loss_fcn(), i.e. criteria = FocalLoss(nn.BCEWithLogitsLoss(), gamma=1.5)."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
):
|
||||
def __init__(self):
|
||||
"""Initializer for FocalLoss class with no parameters."""
|
||||
super().__init__()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue