issue #23

src/model.jl

@@ -160,7 +160,7 @@ struct StepModel{F1,F2}
     costfun::F2
 end
 
-weights(model::StepModel) = reduce(vcat,weights.(model.chains))
+weights(model::StepModel) = Iterators.flatten(weights.(model.chains))
 
 function StepModel(
     chains,loss_weights,truth_uncertain,gamma;

test/runtests.jl

@@ -5,6 +5,7 @@ using Test
 
 @testset "reconstruct gridded data" begin
     include("test_DINCAE_SST.jl")
+    include("test_regL2.jl")
 end
 
 @testset "reconstruct point cloud" begin

test/test_regL2.jl

@@ -0,0 +1,38 @@
+using DINCAE
+
+const F = Float32
+Atype = Array{F}
+
+filename = "avhrr_sub_add_clouds_n10.nc"
+
+if !isfile(filename)
+    download("https://dox.ulg.ac.be/index.php/s/2yFgNMkpsGumVSM/download", filename)
+end
+
+
+data = [
+   (filename = filename,
+    varname = "SST",
+    obs_err_std = 1,
+    jitter_std = 0.05,
+    isoutput = true,
+   )
+]
+data_test = data;
+data_all = [data,data_test]
+
+fnames_rec = [tempname()]
+paramfile = tempname()
+
+losses = DINCAE.reconstruct(
+    Atype,data_all,fnames_rec;
+    epochs = 3,
+    batch_size = 5,
+    enc_nfilter_internal = round.(Int,32 * 2 .^ (0:3)),
+    clip_grad = 5.0,
+    upsampling_method = :nearest,
+    ntime_win = 3,
+    paramfile = paramfile,
+    regularization_L2_beta = 0.001,
+    loss_weights_refine = (0.3,0.7),
+)