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

Made the notebook robust and Trained the model on Aquarium Dataset #80

Merged
merged 1 commit into from
Jun 25, 2024

Conversation

SahilCarterr
Copy link
Contributor

What this PR do?

This pull request addresses issue #79 , aiming to enhance the robustness of Training_Script_example.ipynb and streamline the training process on the Aquarium dataset.

Changes Made

  1. Improved Robustness of Training_Script_example.ipynb:

    • Modified the code to run easily.
    • Refactored code for better readability and maintainability.
  2. Simplified Training on Aquarium Dataset:

    • Added a Dataset.
    • Visualised the Annotated the Images.

Additional Notes:

Feel free to suggest any improvements or additional features that could be included in this example script. Your feedback is highly appreciated!

@zappy586
Copy link

zappy586 commented Jun 23, 2024

Hey! I am coming from 31434. Could you please add a way/ a similar tutorial notebook to train this particular implementation using LoRA? because I am unable to load this particular model using PEFT/LoRA.
Thanks!

@BIGBALLON BIGBALLON merged commit 924bb6c into longzw1997:main Jun 25, 2024
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

3 participants