在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,
}

Answer 1

你试图初始化一个python dictionary。这是你如何做到的：

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