Skip to main content

Simple and powerful pytorch framework.

Project description

EasyTorch

LICENSE PyPI Language grade: Python python lint

English | 简体中文

EasyTorch is an open source neural network framework based on PyTorch, which encapsulates common functions in PyTorch projects to help users quickly build deep learning projects.

:sparkles: Highlight Characteristics

  • :computer: Minimum Code. EasyTorch encapsulates the general neural network training pipeline. Users only need to implement key codes such as Dataset, Model, and training/inference to build deep learning projects.
  • :wrench: Everything Based on Config. Users control the training mode and hyperparameters through the config file. EasyTorch automatically generates a unique result storage directory according to the MD5 of the config file content, which help users to adjust hyperparameters more conveniently.
  • :flashlight: Support All Devices. EasyTorch supports CPU, GPU and GPU distributed training (single node multiple GPUs and multiple nodes). Users can use it by setting parameters without modifying any code.
  • :page_with_curl: Save Training Log. Support logging log system and Tensorboard, and encapsulate it as a unified interface, users can save customized training logs by calling simple interfaces.

:cd: Dependence

OS

Ubuntu 16.04 and later systems are recommended.

Python

python >= 3.6 (recommended >= 3.9)

Miniconda or Anaconda are recommended.

PyTorch and CUDA

pytorch >= 1.4 (recommended >= 1.9). To use CUDA, please install the PyTorch package compiled with the corresponding CUDA version.

Note: To use Ampere GPU, PyTorch version >= 1.7 and CUDA version >= 11.0.

:dart: Get Started

Installation

pip install easy-torch

Initialize Project

TODO

:pushpin: Examples

More examples are on the way

It is recommended to refer to the excellent open source project BasicTS.

:rocket: Citations

BibTex Citations

If EasyTorch helps your research or work, please consider citing EasyTorch. The BibTex reference item is as follows(requires the url LaTeX package).

@misc{wang2020easytorch,
  author =       {Yuhao Wang},
  title =        {{EasyTorch}: Simple and powerful pytorch framework.},
  howpublished = {\url{https://github.com/cnstark/easytorch}},
  year =         {2020}
}

README Badge

If your project is using EasyTorch, please consider put the EasyTorch badge EasyTorch add to your README.

[![EasyTorch](https://img.shields.io/badge/Developing%20with-EasyTorch-2077ff.svg)](https://github.com/cnstark/easytorch)

(Full documentation is coming soon)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

easy_torch-1.3.2-py3-none-any.whl (42.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page