HEmile / storchastic

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

Properly control whether we should take the derivative of the plate weighting #70

Open HEmile opened 4 years ago

HEmile commented 4 years ago

Currently, there is no clear signal whether the plate weight should be detached and where it will be detached and not. This should be made clearer. Taking the derivative over plate weights is currently only used for the expectation (right?)