search

zoogzog / chexnet

Implementation of the CheXNet network (PyTorch)
211 stars 94 forks source link

When the training data is ready for testing,it happens “RuntimeError: CUDA error: out of memory ” #25

Closed stupidBirdFirstFly closed 3 years ago

stupidBirdFirstFly commented 3 years ago

image image Seems to be a problem with the Pytorch version.For me,it's Pytorch-0.4 Has anyone encountered this situation? Thank you very much!

BrQKarkac commented 3 months ago

@stupidBirdFirstFly How to fix the problem? which pytorch version solution?