Skip to content
Snippets Groups Projects
Commit 0a28df3e authored by Merrin Macleod's avatar Merrin Macleod Committed by GitHub
Browse files

Update README.md

parent 55e8c3f7
No related branches found
No related tags found
No related merge requests found
......@@ -168,17 +168,6 @@ python pix2pix.py \
In this mode, image A is the black and white image (lightness only), and image B contains the color channels of that image (no lightness information).
### Exporting the model
You can export the model to be served or uploaded:
```sh
python pix2pix.py \
--mode export \
--output_dir facades_export \
--checkpoint facades_train
```
### Tips
You can look at the loss and computation graph using tensorboard:
......@@ -208,6 +197,17 @@ The test run will output an HTML file at `facades_test/index.html` that shows in
<img src="docs/test-html.png" width="300px"/>
## Exporting
You can export the model to be served or uploaded with `--mode export`. As with testing, you should specify the checkpoint to use with `--checkpoint`.
```sh
python pix2pix.py \
--mode export \
--output_dir facades_export \
--checkpoint facades_train
```
## Code Validation
Validation of the code was performed on a Linux machine with a ~1.3 TFLOPS Nvidia GTX 750 Ti GPU and an Azure NC6 instance with a K80 GPU.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment