Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
ea74ecaa
Commit
ea74ecaa
authored
Jul 15, 2021
by
Philipp Arras
Browse files
Cosmetics
parent
c9687c8a
Changes
1
Show whitespace changes
Inline
Side-by-side
src/minimization/kl_energies.py
View file @
ea74ecaa
...
@@ -416,7 +416,7 @@ def MetricGaussianKL(mean, hamiltonian, n_samples, mirror_samples, constants=[],
...
@@ -416,7 +416,7 @@ def MetricGaussianKL(mean, hamiltonian, n_samples, mirror_samples, constants=[],
def
GeoMetricKL
(
mean
,
hamiltonian
,
n_samples
,
minimizer_samp
,
mirror_samples
,
def
GeoMetricKL
(
mean
,
hamiltonian
,
n_samples
,
minimizer_samp
,
mirror_samples
,
start_from_lin
=
True
,
constants
=
[],
point_estimates
=
[],
start_from_lin
=
True
,
constants
=
[],
point_estimates
=
[],
napprox
=
0
,
comm
=
None
,
nanisinf
=
True
):
napprox
=
0
,
comm
=
None
,
nanisinf
=
True
):
"""Provides the sampled Kullback-Leibler used in geometric Variational
"""Provides the sampled Kullback-Leibler used in geometric Variational
Inference (geoVI).
Inference (geoVI).
...
@@ -487,10 +487,10 @@ def GeoMetricKL(mean, hamiltonian, n_samples, minimizer_samp, mirror_samples,
...
@@ -487,10 +487,10 @@ def GeoMetricKL(mean, hamiltonian, n_samples, minimizer_samp, mirror_samples,
As in MGVI, mirroring samples can help to stabilize the latent mean as it
As in MGVI, mirroring samples can help to stabilize the latent mean as it
reduces sampling noise. But unlike MGVI a mirrored sample involves an
reduces sampling noise. But unlike MGVI a mirrored sample involves an
additional solve of the non-linear transformation. Therefore, when using
additional solve of the non-linear transformation. Therefore, when using
MPI, the mirrored samples also get distributed if enough tasks are
available.
MPI, the mirrored samples also get distributed if enough tasks are
If there are more total samples than tasks, the mirrored
counterparts
available.
If there are more total samples than tasks, the mirrored
try to reside on the same task as their non mirrored partners.
This ensures
counterparts
try to reside on the same task as their non mirrored partners.
that at least the starting position can be re-used.
This ensures
that at least the starting position can be re-used.
See also
See also
--------
--------
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment