I am going to inspect the Neural Model (as it is not done by me). It seems that maybe in some part of the model the mode is changed to train and not to eval.
Even if we set torch.no_grad, if the model is runned in training mode, (model.train()) are the gradients stored?