Nicole

我用现在的pytorch-unet2.0生成了.pth，大概65M，在转engine的时候在70层出现不匹配，请问是有需要改哪里吗

> 好的好的，谢谢你啦

同样想问一下，训练设备是什么，我这边用Xavier Orin做训练（我是骚操作了一点），直接killed

it's worked, thanks!