torch2trt：易于使用的PyTorch到TensorRT转换器
需积分: 46 192 浏览量
2021-02-25
08:22:15
上传
评论
收藏 155KB ZIP 举报
torch2trt：易于使用的PyTorch到TensorRT转换器 （108个子文件） 
group_norm.cpp 8KB interpolate.cpp 8KB plugins.cpp 1KB version-select.css 101B .gitignore 176B conversion.ipynb 5KB live_demo.ipynb 5KB conversion.ipynb 4KB version-select.js 2KB imagenet_labels.json 24KB README.md 7KB reduced_precision.md 6KB jetson_xavier.md 5KB JETSON_XAVIER.md 5KB CONTRIBUTING.md 4KB see_also.md 3KB jetson_nano.md 3KB JETSON_NANO.md 3KB custom_converter.md 2KB basic_usage.md 1KB getting_started.md 1KB LICENSE.md 1KB CHANGELOG.md 636B index.md 446B CONTRIBUTING.md 18B CHANGELOG.md 15B torch2trt.py 19KB interpolate.py 7KB unary.py 7KB instance_norm.py 6KB clamp.py 5KB classification.py 5KB test.py 5KB getitem.py 5KB activation.py 4KB avg_pool.py 4KB mod.py 4KB div.py 3KB sub.py 3KB Conv.py 3KB ConvTranspose.py 3KB floordiv.py 3KB add.py 3KB mul.py 3KB normalize.py 3KB pow.py 3KB ConvTranspose2d.py 3KB split.py 2KB max.py 2KB min.py 2KB view.py 2KB Conv2d.py 2KB mean.py 2KB calibration.py 2KB compare.py 2KB Conv1d.py 2KB sum.py 2KB build.py 2KB utils.py 2KB permute.py 2KB max_pool2d.py 2KB dump_converters.py 2KB prelu.py 2KB transpose.py 2KB chunk.py 2KB group_norm.py 2KB ne.py 2KB batch_norm.py 2KB Linear.py 2KB expand.py 1KB prod.py 1KB __init__.py 1KB softmax.py 1KB relu.py 1KB narrow.py 1KB BatchNorm1d.py 1KB stack.py 1KB segmentation.py 1KB AdaptiveAvgPool2d.py 1KB relu6.py 1KB adaptive_max_pool2d.py 1KB dummy_converters.py 1KB setup.py 1KB pad.py 920B sigmoid.py 916B module_test.py 908B cat.py 871B BatchNorm2d.py 771B identity.py 765B save_load.py 755B tensor.py 649B tanh.py 561B test_contiguous.py 519B LogSoftmax.py 433B __init__.py 395B adaptive_avg_pool2d.py 296B __init__.py 0B __init__.py 0B test.sh 3KB build_docs.sh 115B共 108 条
- 1
- 2
资源评论
