Skip to content

Commit 4a94a74

Browse files
committed
Use new executor in PyTorch RN50 convergence test.
Signed-off-by: Michal Zientkiewicz <[email protected]>
1 parent 68333f3 commit 4a94a74

File tree

1 file changed

+1
-1
lines changed
  • docs/examples/use_cases/pytorch/resnet50

1 file changed

+1
-1
lines changed

docs/examples/use_cases/pytorch/resnet50/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def to_python_float(t):
108108
return t[0]
109109

110110

111-
@pipeline_def
111+
@pipeline_def(experimental_exec_dynamic=True)
112112
def create_dali_pipeline(data_dir, crop, size, shard_id, num_shards, dali_cpu=False, is_training=True):
113113
images, labels = fn.readers.file(file_root=data_dir,
114114
shard_id=shard_id,

0 commit comments

Comments
 (0)