diff --git a/README.md b/README.md index 6f236bd..52a742e 100644 --- a/README.md +++ b/README.md @@ -25,6 +25,11 @@ python run_training.py --model_dir models --head_layer 2 ``` The Script will train a model for each defect type and save it in the `model_dir` Folder. +To enable training on an Nvidia GPU use the `--cuda 1` flag. +``` +python run_training.py --model_dir models --head_layer 2 --cuda 1 +``` + One can track the training progress of the models with tensorboard: ``` tensorboard --logdir logdirs diff --git a/_config.yml b/_config.yml new file mode 100644 index 0000000..f4b4188 --- /dev/null +++ b/_config.yml @@ -0,0 +1,2 @@ +theme: jekyll-theme-cayman +markdown: GFM diff --git a/_includes/head-custom.html b/_includes/head-custom.html new file mode 100644 index 0000000..68632b8 --- /dev/null +++ b/_includes/head-custom.html @@ -0,0 +1 @@ + diff --git a/_layout/default.html b/_layout/default.html new file mode 100644 index 0000000..f70d7a3 --- /dev/null +++ b/_layout/default.html @@ -0,0 +1,42 @@ + + + + + + +{% seo %} + + + + + + + {% include head-custom.html %} + + + + + + +
+ {{ content }} + + +
+ +