Skip to content
GitLab
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in / Register
  • D DeepPavlov
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 18
    • Issues 18
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 22
    • Merge requests 22
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Packages and registries
    • Packages and registries
    • Package Registry
    • Infrastructure Registry
  • Monitor
    • Monitor
    • Incidents
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • DeepPavlov
  • DeepPavlov
  • Merge requests
  • !1284

Feat/torch

  • Review changes

  • Download
  • Email patches
  • Plain diff
Merged Andrei Glinskii requested to merge feat/torch into dev Jul 24, 2020
  • Overview 60
  • Commits 247
  • Pipelines 0
  • Changes 37

Created by: dilyararimovna

  • torch_trainer which is needed to model.train() and model.eval() settings. By default, torch models are set to train mode, so for build_model correct usage we still need to do eval in init of each torch model. We can add try model.eval everywhere in build_model, interact mode, etc but it looks very kostyl'no.
  • base class TorchModel which saves, loads, initializes model on pytorch.
  • torch_classification_model as 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)
Assignee
Assign to
Reviewers
Request review from
Time tracking
Source branch: feat/torch