From 376350d79767546c2069744e80c51def424d12c4 Mon Sep 17 00:00:00 2001 From: Ajinkya Kulkarni <44019576+ajinkya-kulkarni@users.noreply.github.com> Date: Mon, 3 Mar 2025 08:29:41 +0100 Subject: [PATCH] Added flag for checkpoint reading for newer versions of Pytorch --- tools/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/eval.py b/tools/eval.py index 83ad76be8..c213e726c 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -164,7 +164,7 @@ def main(exp, args, num_gpu): ckpt_file = args.ckpt logger.info("loading checkpoint from {}".format(ckpt_file)) loc = "cuda:{}".format(rank) - ckpt = torch.load(ckpt_file, map_location=loc) + ckpt = torch.load(ckpt_file, map_location=loc, weights_only=False) model.load_state_dict(ckpt["model"]) logger.info("loaded checkpoint done.")