Skip to content

get nan loss when training target model in overlooking style module #12

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

Open
lixi92 opened this issue Mar 23, 2024 · 2 comments
Open

get nan loss when training target model in overlooking style module #12

lixi92 opened this issue Mar 23, 2024 · 2 comments

Comments

@lixi92
Copy link

lixi92 commented Mar 23, 2024

get nan loss when training target model in overlooking style module on custom datasets even set learning rate to a very small number

@CHIGIRI19
Copy link

I have the same problem, did you deal with it?

@HeinZingerRyo
Copy link

HeinZingerRyo commented Apr 23, 2024

Hi there. I ran this repo long time ago for my own research. To address the loss=nan issue, I guess it's originally from bbox loss and you may need to check bbox coordinates as it may be out of bounds after +1 or -1. I cannot clearly remember where to modify.

Furthermore, you may need to downgrade to Python 2.7.12 PyTorch 1.0.0 torchvision 0.2.2 as stated in readme. This is deprecated even years ago but I clearly remember that I can't reproduce the results under much higher python and torch version. Good luck to you.

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

3 participants