Skip to content

Instantly share code, notes, and snippets.

@alexiscreuzot

alexiscreuzot/train.sh

Last active August 23, 2018 15:18
Show Gist options
  • Save alexiscreuzot/1b2419f7a00617cda0210f9dc1132de2 to your computer and use it in GitHub Desktop.
Save alexiscreuzot/1b2419f7a00617cda0210f9dc1132de2 to your computer and use it in GitHub Desktop.
Download ZIP
Building a Style Transfer app on iOS using CoreML
Raw
train.sh
python neural_style/neural_style.py train \
--dataset ~/Documents/dataset \
--style-image ~/Documents/images/styles/starry_night.jpg \
--style-size 512 \
--batch-size 2 \
--style-weight 1e10 \
--content-weight 1e5 \
--checkpoint-model-dir ~/Documents/checkpoints \
--save-model-dir ~/Documents/models \
--log-interval 10 \
--cuda 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment