Metadata-Version: 2.0
Name: torcher
Version: 0.0.3
Summary: A pytorch model training util
Home-page: https://github.com/Starangle/torcher
Author: binsu
Author-email: binsu.cs@qq.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown

# torcher

torcher是为pytorch训练模型提供一个类似keras和sklearn中fit方法类似的接口，使用步骤如下
1. 准备好要进行评估的模型（一个`torch.nn.Module`的实例），一个loss函数（返回一个标量`tensor`）以及一系列评价指标（每个函数返回一个标量）；
1. 声明一个Torcher对象（可通过`from torcher import Torcher`导入定义）；
3. 调用fit方法进行拟合。

模块实现见torcher文件夹，test.py提供了一个基于MNIST的简单实例。

目前不支持更换优化器，使用默认的Adam优化器。

