Hi, I tried this with bias set to default value i.e. True.
However, I got this error:
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation. 

When I removed with torch.no_grad(), I got this error:
 RuntimeError: a leaf Variable that requires grad is being used in an in-place operation. 

Any ideas on what else can be tried out?