search

wutianyiRosun / CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]
MIT License
259 stars 51 forks source link

can i do inference using CPU #15

Closed jiaobingle closed 5 years ago

jiaobingle commented 5 years ago

Getting error while running cityscapes_evl.py on Cityscape Data set

when i run with "--gpu 0" , it shows "RuntimeError: CUDA error: out of memory", because of my 2GB GPU. and then, i run without "--cuda " or "--gpu" ,it also looks like "RuntimeError: CUDA error: out of memory",so how can i do test or eval using CPU, thx!

wutianyiRosun commented 5 years ago

@jiaobingle U can refer to the Pytorch official example, and learn how to run the model on the cpu.