Skip to content

Commit 145ec39

Browse files
committed
Merge branch 'dev' into main
2 parents 061fd04 + 0c65eac commit 145ec39

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/losses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def initialize_loss(config, d_out):
1616
return MultiTaskLoss(loss_fn=nn.BCEWithLogitsLoss(reduction='none'))
1717

1818
elif config.loss_function == 'fasterrcnn_criterion':
19-
from examples.models.detection.fasterrcnn import FasterRCNNLoss
19+
from models.detection.fasterrcnn import FasterRCNNLoss
2020
return ElementwiseLoss(loss_fn=FasterRCNNLoss(config.device))
2121

2222
else:

0 commit comments

Comments
 (0)