PyToune:一款类Keras的PyTorch框架

PyToune is a Keras-like framework for PyTorch and handles much of the boilerplating code needed to train neural networks.

官方文档:https://pytoune.org/index.html

可以看到官方文档页面布局也是浓浓的Keras风:

 

手册中主要内容集中在pytoune.framework以及pytoune.framework.callbacks。

pytoune.framework:

中提供了常用的训练所使用到的方法,省去了自己编写训练代码,或在此源码基础上自己做修改。

pytoune.framework.callbacks:
官方说明:

Callbacks are a way to interact with the optimization process. For instance, the ModelCheckpoint callback allows to save the weights of the epoch that has the best “score”,
or the EarlyStopping callback allows to stop the training when the “score” has not gone up for a while, etc. The following presents the callbacks available in PyToune,
but first the documentation of the Callback class shows which methods are available in the callback and what arguments they are provided with.

callbacks用来与优化过程做交互,比如在best acc时保存权重,earlystopping策略等等。

Callbacks分为Callback、Logging、Checkpointing、LR Schedulers四个部分。

 

posted @ 2018-08-04 09:47  木易修  阅读(997)  评论(0编辑  收藏  举报