can't reduce the value of “MainLoss ”

houqb / SPNet

Code for our CVPR2020 paper "Strip Pooling: Rethinking Spatial Pooling for Scene Parsing"

MIT License

387 stars 57 forks source link

can't reduce the value of “MainLoss ” #20

Open zhk779 opened 2 years ago

zhk779 commented 2 years ago

Hi，thanks for your code ,your job is amazing. I meeting a problem when I'm training the model. The value of “MainLoss ” is shaking just in the beginning，can you give me some advice？ I only modify the value of "batch_size" to 2, because I only have one gpu. Thanks.