Skip to main content

dlk: Deep Learning Kit

Project description

A Deep Learning ToolKit

This project is WIP.

Read the Docs

Install

pip install dlk

or 
git clong this repo and do

python setup.py 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.14.tar.gz (87.1 kB view hashes)

Uploaded source

Built Distribution

dlk-0.0.14-py3-none-any.whl (241.3 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page