Merge branch 'dev' into main

p-lambda · Aug 4, 2021 · 145ec39 · 145ec39
2 parents 061fd04 + 0c65eac
commit 145ec39
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: