Adding notebooks to resolve issues #254 and #259 #264
+1,262
−1
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
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