fix import

p-lambda · Aug 4, 2021 · 0c65eac · 0c65eac
1 parent 88ba842
commit 0c65eac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/losses.py b/examples/losses.py
@@ -16,7 +16,7 @@ def initialize_loss(config, d_out):
         return MultiTaskLoss(loss_fn=nn.BCEWithLogitsLoss(reduction='none'))
 
     elif config.loss_function == 'fasterrcnn_criterion':
-        from examples.models.detection.fasterrcnn import FasterRCNNLoss
+        from models.detection.fasterrcnn import FasterRCNNLoss
         return ElementwiseLoss(loss_fn=FasterRCNNLoss(config.device))
 
     else: