RuntimeError: CUDA error: out of memory

danieltan07 / spatialaffinitynetwork

PyTorch Implementation of the paper Learning Affinity via Spatial Propagation Networks

61 stars 12 forks source link

Open fengmeixue opened 5 years ago

fengmeixue commented 5 years ago

I have encountered an error: RuntimeError: CUDA error: out of memory RuntimeError: CUDA error: out of memory Is there any solution?