Needs a lot of GPU memory even 14GB

tangxianfeng / STDN

Code for our Spatiotemporal Dynamic Network

275 stars 98 forks source link

Needs a lot of GPU memory even 14GB #7

Closed Leerw closed 5 years ago

Leerw commented 5 years ago

How much GPU memory is needed to run your main.py? I encountered GPU memory exhausted problem even when I set batch size to 1 and I have totally 14GB GPU memory.