I am imitating this file DANet 10
I return tuple due to

Current PyTorch DataParallel Table is not supporting mutl-gpu loss calculation, which makes the gpu memory usage very in-balance. We address this issue here by doing DataParallel for Model & Criterion. Parallel


I trained this model on lane segmentation dataset , it has train set 1319，I get this resulit :

image.jpg1275×779 160 KB

this image have four images,The bottom right corner of the picture is network input lane label image ,other three images is softmax result of resnet18 model output 。( result_softmax * 255 )

image.png1840×686 102 KB

very faster when i trained model. print training log some abnormalities, some times print two epoch log,some times print one epoch informention。

image.png1546×654 35.7 KB

this train model learning rate and loss
I only change network, used rebuild resnet18,but when I use deeplabv1 model,it can segmentaion lane image
This problem has been bothering me for a few days. I hope you can help me, thank you very much!