Hao Liu

icon indicating an email [email protected]

icon company YanBian University icon location Yanji, China icon location An ordinary graduate student from China who is struggling to publish a paper....

Results 2 comments of Hao Liu

CUDA out of memory

Hi，a new problem has appeared again, even though I set the batch_size to 1, it still out of memory. ![image](https://user-images.githubusercontent.com/20486586/134764222-7936381d-d99f-47a4-82c0-8d9dc0dd7e75.png)

CUDA out of memory

Dear doctor Nirkin, I know where I am wrong, I need to modify the batch-size in /config/train/cityscapes_efficientnet_b1_hyperseg-m.py, instead of writing --batch-size 1 in the terminal. With the help of your...