Use torch.no_grad() instead of inference mode in pytorch_trainer.py

Use torch.no_grad() instead of inference mode in pytorch_trainer.py #3689