在 Pytorch 中加载预训练模型答案

【问题标题】：Load Pre-trained model in Pytorch在 Pytorch 中加载预训练模型
【发布时间】：2019-07-17 23:17:16
【问题描述】：

我目前正在研究GANS，我已经从http://www.cs.columbia.edu/~vondrick/tinyvideo/ 下载了代码和模型，我只需要在其中运行即可获得输出。我已经正确给出了路径，并且我收到错误的代码如下所示。已经正确编写了这些行，但仍然出现语法错误。请帮帮我！

import torch

import torch.legacy.nn

import torchfile

from skimage import io, transform

import torch.nn as nn

import torch.backends.cudnn as cudnn

cudnn.enabled = True

opt = {
  model = 'models/beach/iter63000_net.t7',
  batchSize = 128,
  gpu = 1,
  cudnn = 1,
}

【问题讨论】：

标签： python-2.7 pytorch pre-trained-model

【解决方案1】：

您正在尝试初始化 python dictionary。你就是这样做的：

opt = {
  'model': 'models/beach/iter63000_net.t7',
  'batchSize': 128,
  'gpu':  1,
  'cudnn': 1,
}

【讨论】：

谢谢，我明白了。你能帮我在这行'local noise = torch.Tensor(opt.batchSize, 100):normal():cuda()'
@Rajesh 似乎您正在尝试运行 torch（而不是 pytorch）代码。检查 pytorch 对 legacy 模型的支持以获取更多信息。