YinuoGuo27 commited on
Commit
7aaeb95
·
verified ·
1 Parent(s): 1e9f1d1

Update difpoint/src/models/predictor.py

Browse files
Files changed (1) hide show
  1. difpoint/src/models/predictor.py +0 -2
difpoint/src/models/predictor.py CHANGED
@@ -226,8 +226,6 @@ class OnnxRuntimePredictor:
226
  input_feeds[self.inputs[i].name] = data[i].astype(np.float16)
227
  else:
228
  input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
229
- for key, value in input_feeds.items():
230
- print(f"Input: {key}, Shape: {value.shape}")
231
  results = self.onnx_model.run(None, input_feeds)
232
  return results
233
 
 
226
  input_feeds[self.inputs[i].name] = data[i].astype(np.float16)
227
  else:
228
  input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
 
 
229
  results = self.onnx_model.run(None, input_feeds)
230
  return results
231