We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 061fd04 + 0c65eac commit 145ec39Copy full SHA for 145ec39
1 file changed
examples/losses.py
@@ -16,7 +16,7 @@ def initialize_loss(config, d_out):
16
return MultiTaskLoss(loss_fn=nn.BCEWithLogitsLoss(reduction='none'))
17
18
elif config.loss_function == 'fasterrcnn_criterion':
19
- from examples.models.detection.fasterrcnn import FasterRCNNLoss
+ from models.detection.fasterrcnn import FasterRCNNLoss
20
return ElementwiseLoss(loss_fn=FasterRCNNLoss(config.device))
21
22
else:
0 commit comments