It might be more easy to maintain to use the pytorch version of Gatys method, available as a notebook here: https://github.com/leongatys/PytorchNeuralStyleTransfer
This also might (or might not) help running the code efficiently in V100 cards, for which the current Torch implementation is extremely inefficient.
It might be more easy to maintain to use the pytorch version of Gatys method, available as a notebook here: https://github.com/leongatys/PytorchNeuralStyleTransfer This also might (or might not) help running the code efficiently in V100 cards, for which the current Torch implementation is extremely inefficient.