├── .gitignore ├── LICENSE ├── README.md ├── amsgrad ├── README.md ├── cnn.py ├── mlp.py └── optim.py ├── basic ├── README.md ├── custom_function.py ├── custom_module.py ├── functions.py ├── modules.py └── reference.py ├── binary ├── README.md ├── adam.py ├── cnn.py ├── functions.py ├── mlp.py └── modules.py ├── cffi ├── .gitignore ├── README.md ├── build.py ├── functions │ ├── __init__.py │ └── relu.py ├── modules │ ├── __init__.py │ └── relu.py ├── src │ ├── ext_lib.c │ └── ext_lib.h └── test.py ├── dgc ├── README.md ├── dgc.py └── mlp.py ├── dni ├── README.md └── mlp.py ├── focalloss ├── README.md ├── loss.py └── mnist_mlp.py ├── meprop ├── README.md ├── cnn.py ├── lstm.py ├── mlp.py └── modules.py ├── nvrtc ├── README.md ├── mlp.py ├── relu.py └── requirements.txt ├── rnn ├── .gitignore ├── README.md ├── modules.py ├── test_gru.py ├── test_indrnn.py ├── test_lstm.py ├── test_lstmon.py ├── test_lstmp.py ├── test_mgru.py └── test_rnn.py ├── senet ├── README.md ├── cnn.py └── modules.py └── swish ├── README.md ├── activation.py └── mnist_mlp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/README.md -------------------------------------------------------------------------------- /amsgrad/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/amsgrad/README.md -------------------------------------------------------------------------------- /amsgrad/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/amsgrad/cnn.py -------------------------------------------------------------------------------- /amsgrad/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/amsgrad/mlp.py -------------------------------------------------------------------------------- /amsgrad/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/amsgrad/optim.py -------------------------------------------------------------------------------- /basic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/README.md -------------------------------------------------------------------------------- /basic/custom_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/custom_function.py -------------------------------------------------------------------------------- /basic/custom_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/custom_module.py -------------------------------------------------------------------------------- /basic/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/functions.py -------------------------------------------------------------------------------- /basic/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/modules.py -------------------------------------------------------------------------------- /basic/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/basic/reference.py -------------------------------------------------------------------------------- /binary/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/README.md -------------------------------------------------------------------------------- /binary/adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/adam.py -------------------------------------------------------------------------------- /binary/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/cnn.py -------------------------------------------------------------------------------- /binary/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/functions.py -------------------------------------------------------------------------------- /binary/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/mlp.py -------------------------------------------------------------------------------- /binary/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/binary/modules.py -------------------------------------------------------------------------------- /cffi/.gitignore: -------------------------------------------------------------------------------- 1 | _ext/ 2 | -------------------------------------------------------------------------------- /cffi/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/README.md -------------------------------------------------------------------------------- /cffi/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/build.py -------------------------------------------------------------------------------- /cffi/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cffi/functions/relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/functions/relu.py -------------------------------------------------------------------------------- /cffi/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cffi/modules/relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/modules/relu.py -------------------------------------------------------------------------------- /cffi/src/ext_lib.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/src/ext_lib.c -------------------------------------------------------------------------------- /cffi/src/ext_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/src/ext_lib.h -------------------------------------------------------------------------------- /cffi/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/cffi/test.py -------------------------------------------------------------------------------- /dgc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/dgc/README.md -------------------------------------------------------------------------------- /dgc/dgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/dgc/dgc.py -------------------------------------------------------------------------------- /dgc/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/dgc/mlp.py -------------------------------------------------------------------------------- /dni/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/dni/README.md -------------------------------------------------------------------------------- /dni/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/dni/mlp.py -------------------------------------------------------------------------------- /focalloss/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/focalloss/README.md -------------------------------------------------------------------------------- /focalloss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/focalloss/loss.py -------------------------------------------------------------------------------- /focalloss/mnist_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/focalloss/mnist_mlp.py -------------------------------------------------------------------------------- /meprop/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/meprop/README.md -------------------------------------------------------------------------------- /meprop/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/meprop/cnn.py -------------------------------------------------------------------------------- /meprop/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/meprop/lstm.py -------------------------------------------------------------------------------- /meprop/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/meprop/mlp.py -------------------------------------------------------------------------------- /meprop/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/meprop/modules.py -------------------------------------------------------------------------------- /nvrtc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/nvrtc/README.md -------------------------------------------------------------------------------- /nvrtc/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/nvrtc/mlp.py -------------------------------------------------------------------------------- /nvrtc/relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/nvrtc/relu.py -------------------------------------------------------------------------------- /nvrtc/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | cupy 3 | pynvrtc 4 | -------------------------------------------------------------------------------- /rnn/.gitignore: -------------------------------------------------------------------------------- 1 | *.pkl 2 | -------------------------------------------------------------------------------- /rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/README.md -------------------------------------------------------------------------------- /rnn/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/modules.py -------------------------------------------------------------------------------- /rnn/test_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_gru.py -------------------------------------------------------------------------------- /rnn/test_indrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_indrnn.py -------------------------------------------------------------------------------- /rnn/test_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_lstm.py -------------------------------------------------------------------------------- /rnn/test_lstmon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_lstmon.py -------------------------------------------------------------------------------- /rnn/test_lstmp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_lstmp.py -------------------------------------------------------------------------------- /rnn/test_mgru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_mgru.py -------------------------------------------------------------------------------- /rnn/test_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/rnn/test_rnn.py -------------------------------------------------------------------------------- /senet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/senet/README.md -------------------------------------------------------------------------------- /senet/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/senet/cnn.py -------------------------------------------------------------------------------- /senet/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/senet/modules.py -------------------------------------------------------------------------------- /swish/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/swish/README.md -------------------------------------------------------------------------------- /swish/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/swish/activation.py -------------------------------------------------------------------------------- /swish/mnist_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingKe/pytorch_workplace/HEAD/swish/mnist_mlp.py --------------------------------------------------------------------------------