Added -normalize_weights parameter

ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm

MIT License

834 stars 178 forks source link

Closed ProGamerGov closed 4 years ago

ProGamerGov commented 4 years ago

Added -normalize_weights parameter to normalize content and style weights based on the number of channels in each layer.