Skip to content

Make infer_discrete work with scan#991

Open
fehiepsi wants to merge 11 commits intomasterfrom
infer-discrete-scan
Open

Make infer_discrete work with scan#991
fehiepsi wants to merge 11 commits intomasterfrom
infer-discrete-scan

Conversation

@fehiepsi
Copy link
Member

@fehiepsi fehiepsi commented Apr 3, 2021

Continue of #977. Pair-coded with @eb8680 @fritzo @ordabayevy

TODO

  • fix substituting issues of enum_scan: under enum, latent variable names will be _PREV_z and z while without it, there is only one latent variable names z. One solution is to merge _PREV_z and z in infer_discrete
  • fix adjoint issue with length=10

Follow-up PR: add test for history > 1

Blocked by pyro-ppl/funsor#517


with approx:
approx_factors = funsor.adjoint.adjoint(sum_op, prod_op, log_prob)
approx_factors = tape.adjoint(sum_op, prod_op, log_prob)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this is the line where Scatter and Deltas are introduced

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants