Skip to content

Latest commit

 

History

History
24 lines (15 loc) · 543 Bytes

README.md

File metadata and controls

24 lines (15 loc) · 543 Bytes

NLP-MLOps-Pipeline

End to end MLOps Pipeline for NLP

  • Model: Intent Detection Model Training using Huggingface Trainer API with wandb and Hydra integration

    To train a new model

    python train.py training.max_epochs=4 
    
    
  • DVC: DVC integration for Model Versioning

    To push a new model to remote

    cd dvcfiles
    dvc add ../Notebooks/models/distilroberta-base/ --file clinc_distilroberta.dvc
    dvc push clinc_distilroberta.dvc
    
  • Model Compression: ONNX / Quantization / Pruning