python中forward的参数_python - TypeError：forward（）缺少1个必需的位置参数：“ hidden” - 堆栈内存溢出...

我正在尝试使用PyTorchViz可视化我的GRU模型，但是每次运行此代码都会给我一个错误。

我想要这样的东西

import torch

from torch import nn

from torchviz import make_dot, make_dot_from_trace

model = IC_V6(f.tokens)

x = torch.randn(1,8)

make_dot(model(x), params=dict(model.named_parameters()))

这是我保存数据的班级

class Flickr8KImageCaptionDataset:

def __init__(self):

all_data = json.load(open('caption_datasets/dataset_flickr8k.json', 'r'))

all_data=all_data['images']

self.training_data = []

self.test_data = []

self.w2i = {ENDWORD: 0, STARTWORD: 1}

self.word_frequency = {ENDWORD: 0, STARTWORD: 0}

self.i2w = {0: ENDWORD, 1: STARTWORD}

self.tokens = 2 #END is default

self.batch_index = 0

for data in all_data:

if(data['split']=='train'):

self.training_data.append(data)

else:

self.test_data.append(data)

for sentence in data['sentences']:

for token in sentence['tokens']:

if(token not in self.w2i.keys()):

self.w2i[token] = self.tokens

self.i2w[self.tokens] = token

self.tokens +=1

self.word_frequency[token] = 1

else:

self.word_frequency[token] += 1

def image_to_tensor(self,filename):

image = Image.open(filename)

image = TF.resize(img=image, size=(HEIGHT,WIDTH))

image = TF.to_tensor(pic=image)

image = TF.normalize(image, mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225])

return torch.unsqueeze(image,0)

def return_train_batch(self): #size of 1 always

#np.random.shuffle(self.training_data)

for index in range(len(self.training_data)):

#inde