denshooter/gpu_colorization
1
0
Fork 0
mirror of https://github.com/denshooter/gpu_colorization.git synced 2026-01-21 12:32:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

expand readme

Browse Source
This commit is contained in:
TimNiklasWitte
2022-03-31 14:24:49 +02:00
parent 9a5405a6df
commit 6913e305eb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -55,8 +55,8 @@ batch size = 32
Run `Training.py` to start the training of the model on the `imagenet2012` dataset.
Each epoch the weights are stored into `./saved_models`.
Besides, in `./test_logs` are the corresponding trainings statistics (train and test loss and also a batch of colorized test images) logged.
Note that, the `imagenet2012` dataset must be stored in `./imagenet` as described in this [blog article](https://towardsdatascience.com/preparing-the-imagenet-dataset-with-tensorflow-c681916014ee).
This includes a change of the variable `data_dir` at line 36.
Note that, the `imagenet2012` dataset must be stored in `./imagenet` as described in this [blog article](https://medium.com/analytics-vidhya/how-to-train-a-neural-network-classifier-on-imagenet-using-tensorflow-2-ede0ea3a35ff).
This includes a change of the variable `data_dir` at line 38.
```bash
python3 Training.py