Update pytorch_utils.py

pytorch_utils.py

@@ -54,7 +54,7 @@ def forward(model, x, batch_size):
         pointer += batch_size
 
         with torch.no_grad():
-            with torch.amp.autocast(device_type='cuda', dtype=torch.float16):
+            with torch.amp.autocast(device_type='cuda'):
                 model.eval()
                 batch_output_dict = model(batch_waveform)