Skip to content
GitLab
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in / Register
  • A AutoIntent
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 10
    • Issues 10
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 2
    • Merge requests 2
  • 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
  • AutoIntent
  • Merge requests
  • !93

Refactor/modules dumping and loading

  • Review changes

  • Download
  • Email patches
  • Plain diff
Merged Roman Zlobin requested to merge refactor/modules-dumping-and-loading into dev Jan 11, 2025
  • Overview 16
  • Commits 39
  • Pipelines 0
  • Changes 44

Created by: voorhs

Теперь логика по отгрузке модулей в файловую систему реализована в файле _dump_tools.py. Теперь при написании нового модуля можно не реализовывать методы load и dump, достаточно убедиться что все нужные объекты сохранены в виде атрибутов и имеют допустимый тип (см типы в файле _dump_tools.py

Так то сделано много всего, что стоило разбить на несколько PR:

  • теперь нет объекта VectorIndexClient, поскольку с кешированием от Егора, в нем пропала надобность
  • теперь в классах RerankScorer и DNNCScorer у эмбедера и кросс энкодера разные параметры девайса, батч сайза и все такое
  • настроил приватность и публичность атрибутов модулей
  • теперь Embedder и CrossEncoder отгружаются в файловую систему без сохранения весов трансформеров (ура логи будут весить меньше)

Еще конечно много TODO осталось, но в текущем виде (возможно после устранения ошибок билда документации) этот PR уже можно мерджить чтобы не блокать работу остальных

Assignee
Assign to
Reviewers
Request review from
Time tracking
Source branch: refactor/modules-dumping-and-loading