-
Notifications
You must be signed in to change notification settings - Fork 908
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
base: main
Are you sure you want to change the base?
Conversation
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. |
Hi, @ShreyasJoshi7 👋🏻 Big thanks for your interest in the |
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.
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