Wuvin commited on
Commit
cb68e0f
1 Parent(s): 5d84bc4

update onnx

Browse files
Files changed (1) hide show
  1. scripts/load_onnx.py +6 -6
scripts/load_onnx.py CHANGED
@@ -2,12 +2,12 @@ import onnxruntime
2
  import torch
3
 
4
  providers = [
5
- ('TensorrtExecutionProvider', {
6
- 'device_id': 0,
7
- 'trt_max_workspace_size': 8 * 1024 * 1024 * 1024,
8
- 'trt_fp16_enable': True,
9
- 'trt_engine_cache_enable': True,
10
- }),
11
  ('CUDAExecutionProvider', {
12
  'device_id': 0,
13
  'arena_extend_strategy': 'kSameAsRequested',
 
2
  import torch
3
 
4
  providers = [
5
+ # ('TensorrtExecutionProvider', {
6
+ # 'device_id': 0,
7
+ # 'trt_max_workspace_size': 8 * 1024 * 1024 * 1024,
8
+ # 'trt_fp16_enable': True,
9
+ # 'trt_engine_cache_enable': True,
10
+ # }),
11
  ('CUDAExecutionProvider', {
12
  'device_id': 0,
13
  'arena_extend_strategy': 'kSameAsRequested',