sharvari0b26
/

phase2a

Model card Files Files and versions

sharvari0b26 commited on Jan 9

Commit

c965cd9

·

verified ·

1 Parent(s): 631f4c8

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -65,7 +65,7 @@ if __name__ == "__main__":
     CONF_THRESHOLD = 0.30
     model = RFDETRBase(
-        checkpoint_path="checkpoint_best_ema.pth",
         device="cuda" if torch.cuda.is_available() else "cpu"
     )

     CONF_THRESHOLD = 0.30
     model = RFDETRBase(
+        checkpoint_path="checkpoint_best_total.pth",
         device="cuda" if torch.cuda.is_available() else "cpu"
     )