GitHub repository: wyhwong/classifier-trains
Primary language: Python
Description: This repository aims to provide a primitive tool to finetune state-of-the-art models with PyTorch implementation, similar to Nvidia TAO but with more flexibility in augmentation and models.