深圳幻海软件技术有限公司 欢迎您!

  • pytorch模型保存与加载总结

    pytorch模型保存与加载总结模型保存与加载方式模型保存方式一只存储模型中的参数,该方法速度快,占用空间少(官方推荐使用)model=VGGNet()torch.save(model.state_dict(),PATH)方式二存储整个模型model=VGGNet()torch.save(model

推荐阅读