hanbin's picture
update torch type to float16
b8e1daf verified