Skip to content

Pytorch examples, especially for auto differentiation

Notifications You must be signed in to change notification settings

yjin4-at-uofh/pytorch-huang-fwirnn

 
 

Repository files navigation

pytorch-examples

  1. simple_ad.py: a simple auto differentiation code using Pytorch

  2. simple_lr.py: a simple linear regression code using Pytorch module

  3. fluidsim.py: a fluid simulation using auto differentiation to optimize its velocity model.

  4. sin.py: sin(x) and its gradient verification as cos(x). An example from the Julia paper: A differentiable programming system...

Update Report

@ 02/12/20

  1. Add .gitignore to this project.
  2. Fix the bug in /fwi_dl/Wave1D_AGfunc.py, this bug causes the script could not run on GPU device.

About

Pytorch examples, especially for auto differentiation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%