Moibe commited on
Commit
a49676a
1 Parent(s): b254a4f

Test tensorrt

Browse files
Files changed (1) hide show
  1. funciones.py +3 -2
funciones.py CHANGED
@@ -80,8 +80,9 @@ def mass(input1, input2):
80
  print("Después de los selectores de modo los paths quedaron así:")
81
  print("source_path: ", source_path)
82
  print("target_path: ", target_path)
83
- #FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda.
84
- command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider gpu"
 
85
  print(command)
86
  time.sleep(1)
87
  proc = os.popen(command)
 
80
  print("Después de los selectores de modo los paths quedaron así:")
81
  print("source_path: ", source_path)
82
  print("target_path: ", target_path)
83
+ #FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda o azure?
84
+ #(choose from 'tensorrt', 'cuda', 'cpu')
85
+ command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider tensorrt"
86
  print(command)
87
  time.sleep(1)
88
  proc = os.popen(command)