Open JessonWong opened 1 year ago
Traceback (most recent call last): File "test.py", line 174, in output = output.cpu()
with torch.no_grad(): output= network(content,style)[0] output = output.cpu()
Traceback (most recent call last): File "test.py", line 174, in
output = output.cpu()