Skip to content

Latest commit

 

History

History
28 lines (20 loc) · 535 Bytes

README.md

File metadata and controls

28 lines (20 loc) · 535 Bytes
Raw

tiny-style-transfer

tiny pytorch implementation of neural style transfer.Keep it minimal and one click to run!

Setup

  • install PyTorch,PIL,numpy
  • get the code
git clone https://github.com/ggsonic/tiny-style-transfer
  • one click to run
./run_train.sh

or

python -u tiny.py

Notes

  • use torchvision pretrained models
  • use l1_norm loss to make style weights and content weights in same unit