Update README.md

zhou13 · Mar 9, 2020 · 1d432c1 · 1d432c1
1 parent 49f1b45
commit 1d432c1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -135,6 +135,8 @@ dataset/wireframe.py data/wireframe_raw data/wireframe
 ```
 
 ### Training
+The default batch size assumes your have a graphics card with 12GB video memory, e.g., GTX 1080Ti or RTX 2080Ti. You may reduce the batch size if you have less video memory.
+
 To train the neural network on GPU 0 (specified by `-d 0`) with the default parameters, execute
 ```bash
 python ./train.py -d 0 --identifier baseline config/wireframe.yaml