Блог

Я обучил модель и получил результат с именем model_final.pth.

Я загрузил модель, как показано ниже.

 #load model
model = torch.load('/content/drive/MyDrive/model/model_final.pth')
 

И я последовал нижеприведенным шагам.

 #load sample image
input_image = Image.open("/content/drive/MyDrive/model/g0079.jpg")
display(input_image)

to_tensor = transforms.ToTensor()
input_tensor = to_tensor(input_image)
input_batch = input_tensor.unsqueeze(0)
 

Затем, когда я запускаю приведенный ниже код, я получаю сообщение об ошибке.

 #First attempt at tracing
trace = torch.jit.trace(model, input_batch)
 

Ошибка:

Почему я получаю эту ошибку? Как ты думаешь, что мне следует делать?