torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Published 20 hours ago • verified publisher icon traveller59

How can I convert my own model？

Open ruyijidan opened this issue 5 years ago • 1 comments

model = alexnet(pretrained=True).eval().cuda() model = '/path/to/mymodel' or？

ruyijidan avatar Aug 05 '19 09:08 ruyijidan

model = alexnet(pretrained=True).eval().cuda() it;s regular pytorch model how can I convert my alexnet model ?

ruyijidan avatar Aug 06 '19 02:08 ruyijidan