ImageClassificationTemplate

This is an image classification finetune template using torchvision or timm.

1.prerequisites

python3.7+、gpu

2.installation

pip install -r requirements.txt

3.Support List

Multi-GPU Training Solution
- DistributedDataParallel (DDP) [Built-in function of Pytorch]
Pretrain Model
- totchvision
- timm
- local
Deploy
- onnx(support simplify and optimize)
- mnn
Log
- logger
- tensorboard