Skip to content

About

Assignment 2 for Advanced NLP. Involves fine-tuning and quantization for multi-class classification.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Repository files navigation

hf_emotion_classifier

Assignment 2 for Advanced NLP. Involves fine-tuning and quantization for multi-class classification.

Pretrained weights:

Stored in drive due to its' size being > 25MB https://drive.google.com/drive/folders/1176dlKl5ZOiFk4ahkHex4pc5lEROz4YZ?usp=sharing

File Directories: ./training_scripts directory: Includes script for finetuning, PTQ, QAT and QLoRA. hf_data.py and metrics.py: Contains helper functions to load and summarize metrics respectively. 22161_JiyaSinha_Advance_nlpassignment2.ipynb: Contains the inference codes for the models.

About

Assignment 2 for Advanced NLP. Involves fine-tuning and quantization for multi-class classification.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages