Skip to content

ustcmike/pytorch_workplace

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

33 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Learn pytorch with examples.

Toy example shows how to write customized Function and Module.

Implementation RNN (Vanilla, GRU, LSTM, LSTMP) from scratch. Gradient is clipped to avoid explosion, using pytorch Variable's register_hook function.

BinaryNet with pytorch. Manipulate learning by 1) modifying optimizer (mlp.py) or 2) using param_groups (cnn.py).

Extend pytorch with cffi.

meprop training

Focal Loss

Squeeze and Excitation Networks

Sigmoid-weighted Linear Units (i.e. Swish activation)

Modified (corrected) Adam optimizer.

CUDA extension by compiling CUDA kernels online.

Deep Gradient Compression.

About

Random miniprojects with pytorch.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.3%
  • Other 0.7%