Skip to content
Neural style transfer as a class in PyTorch
Branch: master
Clone or download
Type Name Latest commit message Commit time
Failed to load latest commit information. removed broken optimizer Apr 3, 2017 Update Apr 3, 2017
micromegas_square.jpg initial commit Apr 3, 2017


Neural style transfer as a class in PyTorch

Based on:


StyleTransferNet as a class that can be imported by other scripts

VGG support (this was before pretrained VGG models were available in PyTorch)

Convenience functions for saving intermediate style & content targets for display

Convenience function for examining gram matricies as images

Auto style, content, and product image saving

matplotlib plots of losses over time & record of hyperparameters to keep track of favorable results

You can’t perform that action at this time.