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

Training a teacher network (custom dataset) #87

Open
GuillaumeAnoufa opened this issue Jun 9, 2022 · 1 comment
Open

Training a teacher network (custom dataset) #87

GuillaumeAnoufa opened this issue Jun 9, 2022 · 1 comment

Comments

@GuillaumeAnoufa
Copy link

GuillaumeAnoufa commented Jun 9, 2022

Hello,

Thanks for your work.

I am trying to train SE-SSD on my custom dataset, however the training requires too much gpu memory.
In other issues you suggest using a pre-trained teacher network. (I tried using weights available on your repo but the model and loaded state dict do not match.)

Could you please elaborate on how to train only the teacher network to use it for the actual training after ?
Should I use the CIA-SSD repo ? Or use this repo and change a few things in the config file ?

Edit: I found how to do it (i.e. changing includes in a few init.py files and changing some things in the config file
I think a slightly more detailed readme.md would be welcome however :)

@adnananouzla
Copy link

Hi @GuillaumeAnoufa, could you tell me the steps that you pursued to install spconv?

Thank 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

2 participants