easy-translate / translate_troch2trt.py

Commit History

Fix input type
24a9f47

Iker commited on

Fix bug
745c4a7

Iker commited on

Move batch to device
236bc54

Iker commited on

Test torch2trt
136cf77

Iker commited on