Skip to content

ZJCV/ZCls2

Repository files navigation

Language: 🇺🇸 🇨🇳

«ZCls2» is a more faster classification model training framework



Documentation Status

Table of Contents

Background

After nearly one and a half years of development, ZCls has integrated many training features, includes configuration module, register module, training module, and many model implementations (resnet/mobilenet/senet-sknet-resnest/acbnet-repvgg-dbbnet/ghostnet/gcnet...) and so on. In the development process, it is found that compared with the current excellent classification training framework, such as apex, the training speed of ZCls is not outstanding.

In order to better improve the training speed, we decided to develop a new training framework ZCls2, which is implemented based on apex and provides more friendly and powerful functions. In the preliminary implementation, it can be found that ZCls2 improves the training speed by at least 50% compared with ZCls. More functions are being added.

Installation

See Install

Usage

See Get started

Maintainers

  • zhujian - Initial work - zjykzj

Thanks

Contributing

Anyone's participation is welcome! Open an issue or submit PRs.

Small note:

License

Apache License 2.0 © 2022 zjykzj