test_infer_trt Namespace Reference

Functions
def	parse_args (parser)
def	load_and_setup_model (model_name, parser, checkpoint, amp_run, to_cuda=True)
def	print_stats (measurements_all)
def	main ()

Function Documentation

parse_args()

def test_infer_trt.parse_args

(

parser

)

Parse commandline arguments.

Here is the caller graph for this function:

load_and_setup_model()

def test_infer_trt.load_and_setup_model	(		model_name,
			parser,
			checkpoint,
			amp_run,
			to_cuda = True
	)

Here is the call graph for this function:

Here is the caller graph for this function:

print_stats()

def test_infer_trt.print_stats

(

measurements_all

)

Here is the caller graph for this function:

main()

def test_infer_trt.main

(

)

Launches text to speech (inference).
Inference is executed on a single GPU.

Here is the call graph for this function: