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

Finetuning capabilities? #30

Open
th0mas-codes opened this issue Dec 9, 2023 · 0 comments
Open

Finetuning capabilities? #30

th0mas-codes opened this issue Dec 9, 2023 · 0 comments

Comments

@th0mas-codes
Copy link

Is it possible to finetune this model based on a custom dataset if we meet the COCO panoptic segmentation format standard ?
What I am hoping to achieve is being able to train on a set of very domain specific data I have.

I am currently trying to set up a Colab Notebook where I can try this out.
I have managed to sort my data correctly and generated the following segmentation.png files but I am running into some problems when trying to run the model and i'm unsure where to define the checkpoint .pt provided as a base for the finetuning if so.

I have followed the steps provided in the readme with the setup.
I have gotten so far that when I try and start the training i get this error:

Traceback (most recent call last): File "/content/OpenSeeD/train_net.py", line 466, in <module> launch( File "/usr/local/lib/python3.10/dist-packages/detectron2/engine/launch.py", line 82, in launch main_func(*args) File "/content/OpenSeeD/train_net.py", line 442, in main trainer._trainer.model.module = trainer._trainer.model.module.from_pretrained(cfg.MODEL.WEIGHTS) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1269, in __getattr__ raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'BaseModel' object has no attribute 'module'. Did you mean: 'modules'?

Not sure where to go from here or if what I am trying to do is even possible. Any help or guidance would be very much appreciated!
Really interesting to see what you guys have achieved with this model, great work!

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

1 participant