Skip to main content

Use torch like keras

Project description

torch4keras

licence GitHub release PyPI PyPI - Downloads GitHub stars GitHub Issues contributions welcome

Documentation | Bert4torch | Examples | Source code | build_MiniLLM_from_scratch

1. 下载安装

安装稳定版

pip install torch4keras

安装最新版

pip install git+https://github.com/Tongjilibo/torch4keras.git

2. 功能

  • 简述:抽象出来的Trainer,适用于一般神经网络的训练,仅需关注网络结构代码

  • 特色:进度条展示训练过程,自定义metric,自带Evaluator, Checkpoint, Tensorboard, Logger等Callback,也可自定义Callback

  • 初衷:前期功能是作为bert4torchrec4torch的Trainer

  • 训练:

    torch4keras

3. 快速上手

4. 版本历史

更新日期 版本 版本说明
20240928 v0.2.7 修改ddp的init_process_group;增加一些available函数,增加get_max_memory;DottableDict支持.赋值和嵌套;增加cachedict
20240814 v0.2.6 小修改(增加check_url_available_cached, 修复Timeit)
20240730 v0.2.5 小修改(print_table允许中文, 未安装torch时候仅提醒一次)

更多版本

5. 更新历史:

更多历史

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

torch4keras-0.2.7.tar.gz (60.4 kB view details)

Uploaded Source

File details

Details for the file torch4keras-0.2.7.tar.gz.

File metadata

  • Download URL: torch4keras-0.2.7.tar.gz
  • Upload date:
  • Size: 60.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.0.0 CPython/3.8.8

File hashes

Hashes for torch4keras-0.2.7.tar.gz
Algorithm Hash digest
SHA256 52ebaf0fcbae85ef3a112227b2277034aad730a6841a83f527e908fc555a19b0
MD5 7df6151ff2fc4951775585b66b899c6c
BLAKE2b-256 84764df9151dd978811d843c0c08adce0b2056afe6b008ad9e32d1d7daabd7e2

See more details on using hashes here.

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