Awesome
EasyTorch
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 andTensorboard
, 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 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)