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.