Home

Awesome

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

: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)