Загрузить предварительно обученную модель в 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 .вот как вы это делаете:

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