Skip to content

Commit 58d80e3

Browse files
remove arg
Signed-off-by: Yiheng Wang <[email protected]>
1 parent 976b28e commit 58d80e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: acceleration/fast_inference_tutorial/fast_inference_tutorial.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@
273273
"torch.backends.cudnn.benchmark = True\n",
274274
"torch_tensorrt.runtime.set_multi_device_safe_mode(True)\n",
275275
"device = torch.device(\"cuda:0\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
276-
"train_files = prepare_test_datalist(root_dir, use_compress)\n",
276+
"train_files = prepare_test_datalist(root_dir)\n",
277277
"weights_path = prepare_model_weights(root_dir=root_dir, bundle_name=\"wholeBody_ct_segmentation\")\n",
278278
"trt_model_name = \"model_trt.ts\"\n",
279279
"trt_model_path = prepare_tensorrt_model(root_dir, weights_path, trt_model_name)"

0 commit comments

Comments
 (0)