Commit 5b963678 authored by Amelie Royer's avatar Amelie Royer

Adding CPU mode

parent 857e3589
......@@ -71,7 +71,7 @@ python main.py --nr_gpu 4 --batch_size 16 --test_batch_size 25 --init_batch_size
--mode "eval" --model [path_to_checkpoint]
```
Apply the model on a given input.
Apply the model on a given input (set `--nr_gpu 0` to run in CPU mode).
```bash
python main.py --nr_gpu 1 --batch_size 16 --test_batch_size 25 --init_batch_size 100 \
-c 160 -l 4 --downsample 4 --color lab --dataset imagenet \
......
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