Update trainer.py

2023-10-20 21:55:41 +08:00 · 2023-10-20 21:55:41 +08:00 · ad57a7086c
parent 1e629704c3
commit ad57a7086c
1 changed files with 1 additions and 0 deletions
--- a/maskrcnn_benchmark/engine/trainer.py
+++ b/maskrcnn_benchmark/engine/trainer.py
@ -106,6 +106,7 @@ def do_train(
        try:
            targets = [target.to(device) for target in targets]
            captions = [t.get_field("caption") for t in targets if "caption" in t.fields()]
+            positive_map = positive_map.to(device)
        except:
            pass
        # Freeze language backbone