Update pytorch_utils.py

pytorch_utils.py

@@ -53,10 +53,9 @@ def forward(model, x, batch_size):
         batch_waveform = move_data_to_device(x[pointer : pointer + batch_size], device)
         pointer += batch_size
 
-        with torch.no_grad():
-            with torch.amp.autocast(device_type='cuda'):
-                model.eval()
-                batch_output_dict = model(batch_waveform)
+        with torch.inference_mode():
+            model.eval()
+            batch_output_dict = model(batch_waveform)
 
         for key in batch_output_dict.keys():
             append_to_dict(output_dict, key, batch_output_dict[key].data.cpu().numpy())