leongatys / PytorchNeuralStyleTransfer

Implementation of Neural Style Transfer in Pytorch
MIT License
421 stars 103 forks source link

.data[0] is removed from pytorch . Use .item() as an alternative #9

Closed AlbertPi-Git closed 4 years ago