search

Jakaria08 / EESRGAN

Small-Object Detection in Remote Sensing (satellite) Images with End-to-End Edge-Enhanced GAN and Object Detector Network
GNU General Public License v3.0
279 stars 70 forks source link

python train.py -c config_GAN.json #9

Closed huangxueyan closed 3 years ago

huangxueyan commented 4 years ago

when i run python train.py -c config_GAN.json I meet a problem: RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [3, 64, 3, 3]] is at version 3; expected version 2 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

I surely changed all the Relu(inplace = False), but it doesnot work

Jakaria08 commented 4 years ago

Sorry for the late reply. You should use torch==1.1.0 or torch==1.2. Otherwise, you might face this error.