com Torch.no_grad () se condição

  with torch.set_grad_enabled(not no_grad_condition):
    out=network(input)
Bored Coder