Created by: dilyararimovna
-
torch_trainerwhich is needed tomodel.train()andmodel.eval()settings. By default, torch models are set totrainmode, so forbuild_modelcorrect usage we still need to doevalininitof each torch model. We can add trymodel.evaleverywhere inbuild_model,interactmode, etc but it looks very kostyl'no. - base class
TorchModelwhich saves, loads, initializes model on pytorch. -
torch_classification_modelas an example of Neural Network implementation for classification task. - BERT preprocessors on torch are almost all implemented
- BERT-based classification model on torch (with config example)
- BERT-based ranker model on torch (with config example)
- BERT-based ner model on torch (with config example)
- BERT-based SQuAD model on torch (with config example)
- BERT-based summarization model on torch (with config example)
- files with example models from configs are not all ready (will be uploaded to share later)