/template-text-classification

This is a text classification template using imdb dataset from HuggingFace and training is powered by PyTorch and PyTorch-Ignite.

Primary LanguagePython

Watchers