sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard

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

Open ohaijen opened this issue 1 year ago • 0 comments

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.

ohaijen avatar May 29 '24 17:05 ohaijen