Skip to content

Commit

Permalink
Correct combining of multiple determinants in psiformer with excited …
Browse files Browse the repository at this point in the history
…states.

This was previously corrected for ferminet but not psiformer ansaetze.

PiperOrigin-RevId: 624917909
Change-Id: Id9b6349cd830f6f83cecb40bdf08428870f843f2
  • Loading branch information
jsspencer committed Apr 15, 2024
1 parent 0233831 commit b02ea69
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ferminet/psiformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -445,7 +445,7 @@ def network_apply(
jnp.reshape(orbital, (options.states, -1) + orbital.shape[1:])
for orbital in orbitals
]
return batch_logdet_matmul(*orbitals)
return batch_logdet_matmul(orbitals)
else:
return network_blocks.logdet_matmul(orbitals)

Expand Down

0 comments on commit b02ea69

Please sign in to comment.