speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet
MIT License
772 stars 167 forks source link

About one GPU 1080TI training ? #32

Closed Endless-Hao closed 5 years ago

Endless-Hao commented 5 years ago

I want to ask a question does this code can run in a computer which has just one 1080TI GPU。 When I did it, it said out of memory. And I change the batch_size to just 1, also this problem.