Use inference_mode instead of no_grad (#2804)

ts/torch_handler/base_handler.py

@@ -351,7 +351,7 @@ def inference(self, data, *args, **kwargs):
         Returns:
             Torch Tensor : The Predicted Torch Tensor is returned in this function.
         """
-        with torch.no_grad():
+        with torch.inference_mode():
             marshalled_data = data.to(self.device)
             results = self.model(marshalled_data, *args, **kwargs)
         return results