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

Create training pipeline for SOLOv2 model #3

Open
DomMcOyle opened this issue Nov 21, 2023 · 1 comment
Open

Create training pipeline for SOLOv2 model #3

DomMcOyle opened this issue Nov 21, 2023 · 1 comment
Labels
good first issue Good for newcomers

Comments

@DomMcOyle
Copy link
Owner

As the title suggests, it is required to implement and test the model an the training pipeline SOLOv2 ((paper))

@DomMcOyle DomMcOyle added the good first issue Good for newcomers label Nov 21, 2023
@GiovanniPioDelvecchio
Copy link
Collaborator

image
Not exactly a solvable issue, at least on colab where the torch version is fixed. It turns out that in order to run the solov2 model from the code of the original repository we need to compile some dependencies (using their setup.py) that require torch 1.8, as a feature has been removed in torch 1.11 (see this discussion: open-mmlab/mmdetection3d#1332)

tl:dr; the folder THC/THC.h is missing from recent pytorch versions, this breaks the solov2 legacy code, thus we have to find an alternative

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

No branches or pull requests

2 participants