Skip to main content

dlk: Deep Learning Kit

Project description

A Deep Learning ToolKit

This project is WIP.

Read the Docs


pip install dlk

git clong this repo and do

python install

What is this project do?

  • Provide a templete for deep learning (especially for nlp) training and deploy.
  • Provide parameters search.
  • Provide basic architecture search.
  • Provide some basic modules and models.
  • Provide basic deploy method.

More Feature is Comming

  • Add more documents.

  • Distill structure.

  • Adv training.

  • Add disable tokenizer post process.

  • Predict

    • Complete the main predict code.
    • Test.
    • Convert to TorchScript.
    • Convert to ONNXRT
  • When the high config change the _name of the basic config, the base config should be coverd.
  • One optimizer different para groups use different schedulers.

    • Ref diff_schedule
    • Add loss schedule, get best checkpoint by loss.
  • Support LightGBM, it's maybe not necessary? Will split to another package.

  • Make CRF and more modules which uses the op like for and if to be scriptable(forif
  • Validating LSTM module is scriptable or not?
  • Add UnitTest

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

dlk-0.0.12.tar.gz (85.9 kB view hashes)

Uploaded source

Built Distribution

dlk-0.0.12-py3-none-any.whl (238.6 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page