Bootstrap code for pytorch classification projects.
Clone the repo. First thing that needs checking is the data file. That holds the dataset class. This should be used or updated based on the dataset.
Next thing is to update the model file for specific problem.