Skip to main content

A pytorch model training util

Project description

# 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优化器。

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

torcher-0.0.3.tar.gz (2.4 kB view hashes)

Uploaded source

Built Distribution

torcher-0.0.3-py3-none-any.whl (4.0 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page