From 6156d259289d5cf6ca2fafdea4b1c3221932bc7d Mon Sep 17 00:00:00 2001 From: Tuomas Koskela Date: Thu, 21 May 2020 15:10:38 +0100 Subject: [PATCH] Mean and Var need to be calculated for IO after initial state --- src/TDAC.jl | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/TDAC.jl b/src/TDAC.jl index 599fddfd..d431d2e4 100644 --- a/src/TDAC.jl +++ b/src/TDAC.jl @@ -495,8 +495,12 @@ function tdac(params::tdac_params) end - # Write initial state if params.verbose + # Calculate statistical quantities + @timeit_debug timer "Particle Mean" Statistics.mean!(states.avg, states.particles) + @timeit_debug timer "Particle Variance" states.var .= dropdims(Statistics.var(states.particles; dims=4), dims=4) + + # Write initial state @timeit_debug timer "IO" write_snapshot(states, 0, params) end for it in 1:params.n_time_step @@ -539,7 +543,7 @@ function tdac(params::tdac_params) @timeit_debug timer "Resample" resample!(states.resampled, states.particles, weights) @timeit_debug timer "State Copy" states.particles .= states.resampled - # Calculate statistical values + # Calculate statistical quantities @timeit_debug timer "Particle Mean" Statistics.mean!(states.avg, states.particles) @timeit_debug timer "Particle Variance" states.var .= dropdims(Statistics.var(states.particles; dims=4), dims=4)