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

Train hovernet #76

Merged
merged 14 commits into from
Feb 8, 2021
Merged

Train hovernet #76

merged 14 commits into from
Feb 8, 2021

Conversation

jacob-rosenthal
Copy link
Collaborator

Created a notebook where I train HoVer-Net from scratch.
I'm intending that this will be a reference for users who want to train their own models (how to write training and eval loops in PyTorch, etc.).
It also benchmarks HoVer-Net performance, and now we can use the trained HoVer-Net for downstream tasks.
The model params are too big to put in the repo, so I'll open a new issue for that.

@jacob-rosenthal
Copy link
Collaborator Author

Need to fix failing tests in dev caused by #43 before we can merge this

pathml/ml/hovernet.py Outdated Show resolved Hide resolved
pathml/ml/utils.py Outdated Show resolved Hide resolved
pathml/ml/utils.py Show resolved Hide resolved
pathml/utils.py Show resolved Hide resolved
@jacob-rosenthal jacob-rosenthal merged commit 8bb2970 into dev Feb 8, 2021
@jacob-rosenthal jacob-rosenthal deleted the train-hovernet branch February 8, 2021 20:33
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

Successfully merging this pull request may close these issues.

None yet

2 participants