模型 = YourModelClass(input_size, hidden_​​size, num_layers, output_size) NameError: name \'input_size\' is not defined

Question

import torch
import torch.nn as nn
import torch.optim as optim
answer = input()

class YourModelClass(nn.Module):
    def __init__(self, input_size, hidden_size, num_layers, output_size):
        super().__init__()
        self.lstm = nn.LSTM(input_size, hidden_size, num_layers)
    self.fc = nn.Linear(hidden_size, output_size)

def forward(self, x):
    out, (h_n, c_n) = self.lstm(x)
    out = self.fc(out[:, -1, :])
    return out


model = YourModelClass(input_size, hidden_size, num_layers, output_size)
model.load_state_dict(torch.load('model.ckpt'))
model.eval()


start_index = answer
generated_text = int2char[start_index.item()]


num_generated = 10


with torch.no_grad():
    for i in range(num_generated):
        input_seq = torch.tensor([char2int[generated_text[-1]]]).unsqueeze(0)
        output = model(input_seq)
        output_idx = torch.argmax(output.squeeze()).item()
        generated_text += int2char[output_idx]

print(generated_text)

我无法定义 input_size、hidden_​​size、num_layers、output_size

编辑给我一个错误： 模型 = YourModelClass(input_size, hidden_​​size, num_layers, output_size) NameError：未定义名称“input_size”

Answer 1

您还没有定义任何变量。 尝试这样的事情：

input_size = 100
hidden_size = 128
num_layers = 2
output_size = 50
        
model = YourModelClass(input_size, hidden_size, num_layers, output_size)