03 2023 档案
摘要:
## ONNX导出 ### torch.onnx.export 细解 `torch.onnx.export`中需要的模型实际上是一个`torch.jit.ScriptModule`。而要把普通 PyTorch 模型转一个这样的`TorchScript`模型,有跟踪(trace)和脚本化(script
阅读全文
## ONNX导出 ### torch.onnx.export 细解 `torch.onnx.export`中需要的模型实际上是一个`torch.jit.ScriptModule`。而要把普通 PyTorch 模型转一个这样的`TorchScript`模型,有跟踪(trace)和脚本化(script
阅读全文
posted @ 2023-03-29 16:16
nanmi
摘要:
# CUDA优化  ## 避免Warp分化 ```C++ if (con) { //do somethin
阅读全文
# CUDA优化  ## 避免Warp分化 ```C++ if (con) { //do somethin
阅读全文

浙公网安备 33010602011771号