Jeston运行mmseg需要pytorch 1.11,jetson版的pytorch 2.0和2.1不支持分布式训练,mmseg中关于torch.dist的部分会报错,支持分布式训练的最到版本的torch是1.11
目前看样子还是无法将带transformer的图像分割模型转为tensorRT的模型,有些attention中的算子在tensorRT中还不支持。
Jeston运行mmseg需要pytorch 1.11,jetson版的pytorch 2.0和2.1不支持分布式训练,mmseg中关于torch.dist的部分会报错,支持分布式训练的最到版本的torch是1.11
目前看样子还是无法将带transformer的图像分割模型转为tensorRT的模型,有些attention中的算子在tensorRT中还不支持。