Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The low mIoU of validation dataset #17

Open
Huhaowen0130 opened this issue Jun 21, 2021 · 1 comment
Open

The low mIoU of validation dataset #17

Huhaowen0130 opened this issue Jun 21, 2021 · 1 comment

Comments

@Huhaowen0130
Copy link

I have trained the model for 100 epochs on SUNRGBD with pretrained ResNet34 on Imagenet, and the best mIoU is still 10%. Is it normal?

My batchsize is 16, and the other hyper-parameters are set as default.

@mona0809
Copy link
Collaborator

Have you reduced the maximum epochs with the args or have you just stopped the training after 100 epochs? Our learning rate scheduler reaches the highest learning rate only after 10% of the total epochs. Since we use 500 epochs, the highest learning rate is reached after 50 epochs and then only slowly decreases. So maybe the learning rate is too high.
Still, the best mIoU should be higher after training for 100 epochs. Due to limited GPU memory, we never trained with batchsize 16. Do you have the same problems when training with batchsize 8?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants