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

Adding notebooks to resolve issues #254 and #259 #264

Open
wants to merge 10 commits into
base: main
Choose a base branch
from

Conversation

ShreyasJoshi7
Copy link

Description

I have added a notebook which demonstrates the training for an image classifier using DeiT. The pipeline downloads data from roboflow, specifically the Tumor-Classification-1 dataset for training, validation and testing. This PR also contains a trainging notebook for DCGAN.
DeiT --> train-DeiT-image-classifier-on-custom-data.ipynb
DCGAN --> train-dcgan-on-custom-data.ipynb

dependencies: Roboflow (for dataset download)

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • [ x] New feature (non-breaking change which adds functionality)
  • [ x] This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

I have tested this code on the test dataset that i downloaded from Roboflow.

Any specific deployment considerations

Nope. just follow the instructions and explainations in the notebooks and you should be good to go!

Docs

  • [ X] Docs updated? What were the changes: I have updated the models tables in the readme file.

@CLAassistant
Copy link

CLAassistant commented May 5, 2024

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
2 out of 3 committers have signed the CLA.

✅ SkalskiP
✅ ShreyasJoshi7
❌ Shreyas Joshi


Shreyas Joshi seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

@SkalskiP
Copy link
Collaborator

SkalskiP commented May 6, 2024

Hi, @ShreyasJoshi7 👋🏻 Big thanks for your interest in the notebooks repository and the time you've dedicated to this PR. Is this PR related to any currently open issues?

@ShreyasJoshi7
Copy link
Author

hi @SkalskiP, No this PR is not related to any other issues. Only related to #254 and #259

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