pytorch下模型和数据位置的处理

字数统计: 58字   |   阅读时长: 1分
阅读量

整理实用的代码。

查询数据位置

模型参数

1
print(next(model.parameters()).is_cuda)

tensor数据

1
print(data.device)

转移参数位置

1
2
model.load_state_dict(torch.load('Epoch99.pth'))
model.to(torch.device("cuda:1"))  # 将模型移到1号gpu上

扫一扫，分享到微信

微信分享二维码

tag:

    缺失模块。
    1、请确保node版本大于6.2
    2、在博客根目录（注意不是yilia根目录）执行以下命令：
    npm i hexo-generator-json-content --save

    3、在根目录_config.yml里添加配置： 

      jsonContent:
    meta: false
    pages: false
    posts:
      title: true
      date: true
      path: true
      text: false
      raw: false
      content: false
      slug: false
      updated: false
      comments: false
      link: false
      permalink: false
      excerpt: false
      categories: false
      tags: true

有待补充<br>