Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 24bd2ca

Browse files
author
DEKHTIARJonathan
committed
[Benchmarking-Py] W&D Temporary Fix in Dataloader
1 parent 7481427 commit 24bd2ca

File tree

1 file changed

+4
-1
lines changed
  • tftrt/benchmarking-python/nvidia_examples/wide_deep_tf2

1 file changed

+4
-1
lines changed

tftrt/benchmarking-python/nvidia_examples/wide_deep_tf2/infer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,10 @@ def preprocess_model_inputs(self, data_batch):
180180

181181
x, y = data_batch
182182

183-
y = {"y": y, "display_ids": x.pop('display_id')}
183+
if not self._args.use_synthetic_data: # ISSUE WAR
184+
y = {"y": y, "display_ids": x.pop('display_id')}
185+
else:
186+
x.pop('display_id', None)
184187

185188
return x, y
186189

0 commit comments

Comments
 (0)