|
@@ -341,7 +341,7 @@ def main(args):
|
|
|
model_ema = utils.ExponentialMovingAverage(model_without_ddp, device=device, decay=1.0 - alpha)
|
|
model_ema = utils.ExponentialMovingAverage(model_without_ddp, device=device, decay=1.0 - alpha)
|
|
|
|
|
|
|
|
if args.resume:
|
|
if args.resume:
|
|
|
- checkpoint = torch.load(args.resume, map_location="cpu", weights_only=True)
|
|
|
|
|
|
|
+ checkpoint = torch.load(args.resume, map_location="cpu", weights_only=False)
|
|
|
model_without_ddp.load_state_dict(checkpoint["model"])
|
|
model_without_ddp.load_state_dict(checkpoint["model"])
|
|
|
if not args.test_only:
|
|
if not args.test_only:
|
|
|
optimizer.load_state_dict(checkpoint["optimizer"])
|
|
optimizer.load_state_dict(checkpoint["optimizer"])
|
