CS-Net
CS-Net copied to clipboard
fix a runtime error in later pytorch versions
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
Changing the out += residual
to out = out + residual
does fix the issue.