Update Image training code for downstream datasets (iWildcam, transfer datasets)

neuralmagic / sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

Apache License 2.0

2.07k stars 148 forks source link

Open ohaijen opened 5 months ago

ohaijen commented 5 months ago

Update trochvision/train.py to accomodate:

iWildcam. This code would be read through the ImageFolder format, but requires different resizes and crops.
Transfer datasets (Aircraft, DTD, Flowers) - these would be read through the TorchVision API and cached in /tmp. These don't need any new transforms, just the class loading.