cityscapes训练的问题

Reagan1311 / DABNet

Depth-wise Asymmetric Bottleneck for Real-time Semantic Segmentation (BMVC2019)

https://github.com/Reagan1311/DABNet

MIT License

140 stars 36 forks source link

Closed buzhiqimeiliuqiangdong closed 5 years ago

buzhiqimeiliuqiangdong commented 5 years ago

camvid数据集没有问题，但是当我训练cityscapes数据集，debug时‘ProbOhemCrossEntropy2d(nn.Module)’出现以下问题由于显存不够，我用的batchsize是2，希望大佬有时间时能回复下，谢谢您

Reagan1311 commented 5 years ago

代码在我电脑上没有这个问题诶，你把.long()改成.double()试试

buzhiqimeiliuqiangdong commented 5 years ago

依然没有用，我尝试把cityscape数据集的’ProbOhemCrossEntropy2d‘损失函数换成’CrossEntropyLoss2d‘是可以训练的，请问这两个损失函数对结果有多大的影响？谢谢您

Reagan1311 commented 5 years ago

影响不大，大概在1%以内