2021-3-19

PyTorch模型保存与加载
部署运行你感兴趣的模型镜像
  1. 看论文要注意论文未解决的问题和可能的解决方案,今天深刻体会了。

  2. top 命令可以查看系统当前的pid, user等信息,可以用来kill -9 强制停止程序

  3. .pth 和 .pth.tar都可以直接load

  4. torch.save() 和 torch.load()
    一、save()
    先建立一个字典,保存三个参数:
    state = {‘net’:model.state_dict(), ‘optimizer’:optimizer.state_dict(), ‘epoch’:epoch}
    2.调用torch.save():torch.save(state, dir)
    其中dir表示保存文件的绝对路径+保存文件名,如’/home/qinying/Desktop/modelpara.pth’
    二、load()
    当你想恢复某一阶段的训练(或者进行测试)时,那么就可以读取之前保存的网络模型参数等。
    cpu->cpu, gpu->gpu 不用加maplocation
    gpu->cpu map_location=‘cpu’
    checkpoint = torch.load(dir,map_location=torch.device(‘cpu’))
    model.load_state_dict(checkpoint[‘net’])
    optimizer.load_state_dict(checkpoint[‘optimizer’])
    start_epoch = checkpoint[‘epoch’] + 1

  5. 如果多卡跑程序,一个进程结束,另一个不一定结束,这时候会占用现存。要手动kill

  6. gpustat 插件可以动态查看gpu使用情况,vim ~/.bashrc
    alias mn = “watch --color -n 2 gpustat --color”
    退出来后source ~/.bashrc即可

  7. 多头注意力,每个头都还有一个权重。

  8. pytorch康复训练

  9. einops 学习一下

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值