Torch.no_grad() for Java

What is the equivalent for doing torch.no_grad() in Java?
Like in C++ it is torch::NoGradGuard


You mean on mobile? Or do you refer to another Java binding?
IIRC it does not support training, so it is running in no_grad all the time.

I mean running it on mobile.

Java binding works for forward only and auto grad is disabled all the time.

It has torch::autograd::AutoGradMode g{false} before every forward operation.

1 Like