Commit ea4eb561 authored by Amélie Royer's avatar Amélie Royer Committed by GitHub

Update README.md

parent d4f0dc58
......@@ -79,7 +79,7 @@ python main.py --nr_gpus 1 -c 160 -l 4 --downsample 2 --color lab --dataset cifa
--mode "demo" --model [path_to_checkpoint .ckpt] --input [path to grayscale image]
```
with the model pretrained on CIFAR
with the model pretrained on Imagenet
```bash
python main.py --nr_gpus 1 -c 160 -l 4 --downsample 4 --color lab --dataset imagenet \
--mode "demo" --model [path_to_checkpoint .ckpt] --input [path to grayscale image]
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment