Open karlcao opened 4 years ago
can not from torch.utils.ffi import _warp
你好,请问单张gpu是不是可以直接用nn.BatchNorm2d,然后DataParallelModel(deeplab),,DataParallelCriterion(criterion)这两句都可以不要对吧
还有你这个代码当中的模型中输出两个值下,x和dsn,dsn用于辅助损失函数,像pspnet那样,但计算loss时并没有把dsn考虑进去
can not from torch.utils.ffi import _warp
how do you solve the problem?I have the same problem with you.
1.find the following sentence: from torch.utils.ffi import create_extension change it to : from torch.utils.cpp_extension import BuildExtension
这个好像并不能解决问题
I follow the guidance, but pytorch1.0 do not support torch.utils.ffi. I can not run the code