diyiiyiii / StyTR-2

StyTr2 : Image Style Transfer with Transformers
352 stars 64 forks source link

AttributeError: 'tuple' object has no attribute 'cpu' #22

Open JessonWong opened 1 year ago

JessonWong commented 1 year ago

Traceback (most recent call last): File "test.py", line 174, in output = output.cpu()

proboscis commented 1 year ago
with torch.no_grad():
    output= network(content,style)[0]
    output = output.cpu()