search

HEmile / storchastic

Stochastic Automatic Differentiation library for PyTorch.
GNU General Public License v3.0
180 stars 5 forks source link

when will you include ARM? #89

Closed zhaoguangyuan123 closed 3 years ago

zhaoguangyuan123 commented 3 years ago

thank you

HEmile commented 3 years ago

Hi there! A preliminary implementation of ARM is already available at the DiCE branch: https://github.com/HEmile/storchastic/blob/dice/storch/method/arm.py I still have to polish it though, and include ARSM.

I'd recommend DisARM too (implemented in that same file), it has provably lower variance.

zhaoguangyuan123 commented 3 years ago

thank you for your kind reply. i notice it now.