torch4keras | Keras Ecosystem Directory

Bot releases are visible (Hide)

torch4keras - 稳定版本v0.0.9

Published by Tongjilibo over 1 year ago

增加auto_set_cuda_devices自动选择显卡，增加log_info，log_warn, log_error等小函数

torch4keras - 稳定版本v0.0.8

Published by Tongjilibo over 1 year ago

增加EmailCallback和WandbCallback, 增加AccelerateTrainer和DeepSpeedTrainer, grad_accumulation_steps内算一个batch，修改Trainer中部分成员函数

torch4keras - 稳定版v0.0.7.post3

Published by Tongjilibo over 1 year ago

修复保存scheduler

torch4keras - 稳定版本v0.0.7.post2

Published by Tongjilibo over 1 year ago

Checkpoint Calback增加保存scheduler, save_weights可自行创建目录，Logger, Tensorboard模块加入lr, 修改predict和add_trainer

torch4keras - 稳定版本v0.0.7

Published by Tongjilibo over 1 year ago

独立出callbacks.py文件, fit允许输入形式为字典，load_weights支持list输入，save_weights支持仅保存可训练参数

torch4keras - 稳定版本v0.0.6

Published by Tongjilibo over 1 year ago

增加hf的accelerator测试用例, ddp需要外部控制执行callback, 混合精度支持bf16, 增加add_trainer方法

torch4keras - 稳定版本v0.0.5

Published by Tongjilibo almost 2 years ago

增加Summary的Callback, 增加Tqdm的进度条展示，保留原有BaseModel的同时，增加Trainer(不从nn.Module继承), 从bert4torch的snippets迁移部分通用函数

torch4keras - 稳定版本v0.0.4

Published by Tongjilibo almost 2 years ago

为callback增加on_train_step_end方法, 修复BaseModel(net)方式的bug

torch4keras - 修复版本v0.0.3.post

Published by Tongjilibo almost 2 years ago

修复DDP下只有master_rank打印

torch4keras - 稳定版本v0.0.3

Published by Tongjilibo almost 2 years ago

参考Keras修改了callback的逻辑

torch4keras - 稳定版本v0.0.2

Published by Tongjilibo almost 2 years ago

增加Checkpoint, Evaluator等自带Callback, 修改BaseModel(net)方式

torch4keras - v0.0.1

Published by Tongjilibo about 2 years ago

初始版本，从bert4torch中剥离出来的普通Trainer，支持单机多卡，混合精度，梯度裁剪，梯度累积等

Package Rankings

Top 14.23% on Pypi.org

Top 6.68% on Proxy.golang.org

Badges

Extracted from project README

Related Projects

awesome-tensorlayer

A curated list of dedicated resources and applications

06 May 2018 266

tensorspace

Neural network 3D visualization framework, build interactive and intuitive model in browsers, sup...

22 Jul 2018 5,041

Keras-TextClassification

中文长文本分类、短句子分类、多标签分类、两句子相似度（Chinese Text Classification of Keras NLP, multi-label classify, or sen...

13 Jun 2019 1,761

tensorflow-tutorial

TensorFlow and Deep Learning Tutorials

25 Aug 2016 738

torchtools

A High-Level training API on top of PyTorch

06 Sep 2017 17

keras4torch

A compatible-with-keras wrapper for training PyTorch models✨

10 Oct 2020 30

Deep-Learning-in-Production

In this repository, I will share some useful notes and references about deploying deep learning-b...

03 May 2018 4,294

langml

A Keras-based and TensorFlow-backend NLP Models Toolkit.

03 Nov 2021 11

tfkg

Create, train, and save Tensorflow Keras models all in Golang

06 Dec 2021 49

toolkit4nlp

transformers implement (architecture, task example, serving and more)

30 Jun 2020 97

Deep_Learning_TensorFlow2_Examples

深度学习TensorFlow2入门指南。Deep Learning TensorFlow2 API and Examples with Python3 and Jupyter Notebook.

27 Feb 2024 98

torchkeras

Pytorch❤️ Keras 😋😋

21 Jun 2020 1,715

bert-for-tf2

A Keras TensorFlow 2.0 implementation of BERT, ALBERT and adapter-BERT.

22 May 2019 803

keras-aug

A library that includes pure TF/Keras preprocessing and augmentation layers, providing support fo...

10 Apr 2023 8

torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

12 Aug 2019 568