Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
aba5df20
Commit
aba5df20
authored
Sep 11, 2018
by
Reimar H Leike
Browse files
introducing constants_sampling for MPI paralllel KL
parent
77ef8995
Changes
1
Hide whitespace changes
Inline
Side-by-side
nifty5/minimization/kl_energy.py
View file @
aba5df20
...
...
@@ -45,16 +45,20 @@ class KL_Energy_MPI(Energy):
h
,
nsamp
,
constants
=
[],
constants_samples
=
None
,
_samples
=
None
,
want_metric
=
False
):
super
(
KL_Energy_MPI
,
self
).
__init__
(
position
)
self
.
_h
=
h
self
.
_nsamp
=
nsamp
self
.
_constants
=
constants
if
constants_samples
is
None
:
constants_samples
=
constants
self
.
_constants_samples
=
constants_samples
self
.
_want_metric
=
want_metric
if
_samples
is
None
:
lo
,
hi
=
_shareRange
(
nsamp
,
ntask
,
rank
)
met
=
h
(
Linearization
.
make_var
(
position
,
True
)).
metric
met
=
h
(
Linearization
.
make_
partial_
var
(
position
,
constants_samples
,
True
)).
metric
_samples
=
[]
for
i
in
range
(
lo
,
hi
):
np
.
random
.
seed
(
i
)
...
...
@@ -73,7 +77,7 @@ class KL_Energy_MPI(Energy):
def
at
(
self
,
position
):
return
KL_Energy_MPI
(
position
,
self
.
_h
,
self
.
_nsamp
,
self
.
_constants
,
self
.
_samples
,
self
.
_want_metric
)
self
.
_constants_samples
,
self
.
_samples
,
self
.
_want_metric
)
@
property
def
value
(
self
):
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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