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

Commit 4ce2ed6

Browse files
authored
remove extra argument from optimization profile helper (#3)
1 parent 0ece6bd commit 4ce2ed6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

classification/imagenet/onnx_to_tensorrt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def main():
197197
# Add optimization profiles
198198
batch_sizes = [1, 8, 16, 32, 64]
199199
inputs = [network.get_input(i) for i in range(network.num_inputs)]
200-
opt_profiles = create_optimization_profiles(builder, inputs, args.explicit_batch, batch_sizes)
200+
opt_profiles = create_optimization_profiles(builder, inputs, batch_sizes)
201201
add_profiles(config, inputs, opt_profiles)
202202
# Implicit Batch Network
203203
else:

0 commit comments

Comments
 (0)