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.")