jiwoon-ahn / psa

Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentation, CVPR 2018
MIT License
380 stars 62 forks source link

The train speed of classification network #20

Open TSKongLingwei opened 5 years ago

TSKongLingwei commented 5 years ago

Hello, When I train the classification Network, i find it is too slow. It almost take 1 second for each step with batch-size 4. Data loader is the most Time-consuming part. It is also very slow when calculate the loss with x and label. The forward part and the background part are quick. I test it on pytorch0.4.0 and pytorch 1.0. The results are similar. Is this normal? Thank you very much.