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