Thanks for the reply. However, based one of your previous rely from other post (Why relu(inplace=True) does not give error in official resnet.py but it gives error in my code?), it says that the conv operation need it’s output to be able to compute the backward pass. Not sure if I missed anything.
Thank you again!