Here I have written a model and a pre-trained model for a classification task.
Primary LanguageJupyter Notebook