diff --git a/models/epoch_20_all.pth b/models/epoch_20_all.pth new file mode 100644 index 0000000000000000000000000000000000000000..d37641cd61bc0b94545ae9b3586fdba5bdf42105 --- /dev/null +++ b/models/epoch_20_all.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c27b44360b84e980f33d322eaa6a55acec756ee70b7051c1f0a574aca529e3a3 +size 1129381738 diff --git a/new_eval_all.py b/new_eval_all.py index b835ffb750283d2167387d34c16d1224024b5d66..c7c31631ecced3d1666c2ea63ec5076849ec5861 100644 --- a/new_eval_all.py +++ b/new_eval_all.py @@ -242,7 +242,7 @@ def evaluate(IMAGE_DIR, PREDICTION_FILE): if fp16_cfg is not None: wrap_fp16_model(model) - checkpoint = load_checkpoint(model,"models/epoch_19_all.pth", map_location='cpu') + checkpoint = load_checkpoint(model,"models/epoch_20_all.pth", map_location='cpu') json_file = open("data/edited_val_r3.json")