Hi Alban, sorry to bother you again. I have tried checking the requires_grad of all the variables involved and they are all fine except c. Although, all the consequent variables that make use of c have requires_grad set to True



 _zero:

c = self.var2[idxs[:,:K]]
# here c.requires_grad == False



Do you know why?