liutinglt / CE2P

214 stars 41 forks source link

Different result of mean IU under almost the same circumstance except different batch_size #43

Open tjpulkl opened 4 years ago

tjpulkl commented 4 years ago

Hi: Could you tell me why I get the following results when I download and run your code? the bigger of the batch_size, the lower mean IU value Batch_size Input_size Docker ID Mean IU 4 384,384 cd0 53.808 8 384,384 cd0 51.93 12 384,384 cd0 51.059 14 384,384 cd0 50.71 16 384,384 d9a 50.78 24 384,384 d9a 49.34

Thank you!

YU-0626 commented 3 years ago

4 384,384 Could you tell me about your learning rate,when your batch_size=4 input_size=384?thanks!

tjpulkl commented 3 years ago

LR=1e-3. if we set LR=7.0e-3, you can set batch_size=24, and get the result above 53.1