Skip to content

Commit ada8957

Browse files
authored
Update README.md
1 parent 49fcd2b commit ada8957

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,15 +87,15 @@ docker pull mgkang/studiogan:latest
8787
This is my command to make a container named "studioGAN".
8888

8989
Also, you can use port number 6006 to connect the tensoreboard.
90-
```
90+
```console
9191
docker run -it --gpus all --shm-size 128g -p 6006:6006 --name studioGAN -v /home/USER:/root/code --workdir /root/code mgkang/studiogan:latest /bin/bash
9292
```
9393

9494

9595
## Quick Start
9696

9797
* Train (``-t``) and evaluate (``-e``) the model defined in ``CONFIG_PATH`` using GPU ``0``
98-
```
98+
```console
9999
CUDA_VISIBLE_DEVICES=0 python3 src/main.py -t -e -c CONFIG_PATH
100100
```
101101

0 commit comments

Comments
 (0)