Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train a new model? #88

Open
linrio opened this issue Apr 14, 2017 · 1 comment
Open

How to train a new model? #88

linrio opened this issue Apr 14, 2017 · 1 comment

Comments

@linrio
Copy link

linrio commented Apr 14, 2017

In the model file, It has two exampled model. How two train a new kind of model? Like starry_night etc?

@HappyCoderMan
Copy link

Use the train.py script to build a new model.

Template for train.py looks like this:
python train.py -s <style_image_path> -d <training_dataset_path> -g <use_gpu ? gpu_id : -1>

A specific example could look like this:
python train.py -s ./StarryNight.jpg -d ./train2014 -o /StarryNight/StarryNight -g 0

That specific example assumes you extracted the Microsoft COCO dataset to the train2014 folder and want to use GPU with ID 0. You'll want to create the StarryNight output folder too, else you'll get an error while the model is written to disk.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants