How can we benefit afterwards from saving both model and [args.arch, model.state_dict()] ?
# Final model
    torch.save(model, MODEL_DIR / f"final_{args.fold}.pth")
    # Failover (args + state dict)
    torch.save(
        [args.arch, model.state_dict()],
        MODEL_DIR / f"failover_{args.arch}_{args.fold}.pth"