Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Chris Jewell
covid19uk
Commits
73fcac5c
Commit
73fcac5c
authored
Mar 10, 2021
by
Chris Jewell
Browse files
Added option
parent
5630746b
Changes
1
Hide whitespace changes
Inline
Side-by-side
covid/tasks/inference.py
View file @
73fcac5c
...
...
@@ -54,7 +54,7 @@ def _get_window_sizes(num_adaptation_steps):
return
first_window_size
,
slow_window_size
,
last_window_size
@
tf
.
function
@
tf
.
function
(
jit_compile
=
False
)
def
_fast_adapt_window
(
num_draws
,
joint_log_prob_fn
,
...
...
@@ -112,7 +112,7 @@ def _fast_adapt_window(
return
draws
,
trace
,
step_size
,
weighted_running_variance
@
tf
.
function
@
tf
.
function
(
jit_compile
=
False
)
def
_slow_adapt_window
(
num_draws
,
joint_log_prob_fn
,
...
...
@@ -181,7 +181,7 @@ def _slow_adapt_window(
)
@
tf
.
function
# (experimental
_compile=
Tru
e)
@
tf
.
function
(
jit
_compile
=
Fals
e
)
def
_fixed_window
(
num_draws
,
joint_log_prob_fn
,
...
...
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