python-2.7 - 在 Pytorch 中加载预训练模型

Question

我目前正在研究 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.7pytorchpre-trained-model

Answer 1

您正在尝试初始化 python字典。这就是你的做法：

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