Как мне решить эту ошибку? AttributeError: у объекта builtin_function_or_method нет атрибута data

Question

Я пытаюсь реализовать передачу нейронного стиля, используя код в этой статье. Когда я запускаю main.py, в заголовке моего вопроса появляется ошибка, относящаяся к строке 12. Я почти уверен, что это как-то связано с torch.randn(input.data.size()), но я не знаю, что с этим делать. Я все еще начинаю с машинным обучением, и я все еще изучаю вещи о Python. Любая помощь будет оценена.

from StyleCNN import *
from utils import *

# CUDA Configurations
dtype = torch.cuda.FloatTensor if torch.cuda.is_available() else torch.FloatTensor

# Content and style
style = image_loader("style.jpg").type(dtype)
content = image_loader("content.jpg").type(dtype)

pastiche = image_loader("content.jpg").type(dtype)
pastiche.data = torch.randn(input.data.size()).type(dtype)

num_epochs = 31


def main():
    style_cnn = StyleCNN(style, content, pastiche)

    for i in range(num_epochs):
        pastiche = style_cnn.train()

        if i % 10 == 0:
            print("Iteration: %d" % (i))

            path = "outputs/%d.png" % (i)
            pastiche.data.clamp_(0, 1)
            save_image(pastiche, path)


main()