easy-translate / translate_troch2trt.py

Commit History

Fix bug
dacb273

Iker commited on

Fix bug
a911872

Iker commited on

Move attention mask to device
ab7a062

Iker commited on

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