From d68122c5b04066492e3e2434731a2c46d228f923 Mon Sep 17 00:00:00 2001 From: BulatVakhitov Date: Tue, 16 Jul 2024 15:31:53 +0000 Subject: [PATCH] small fix --- batchflow/models/torch/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py index e7f213f0c..16fc63177 100755 --- a/batchflow/models/torch/base.py +++ b/batchflow/models/torch/base.py @@ -1529,7 +1529,7 @@ def _tile(self, array, microbatch_size): def aggregate_microbatches(self, outputs_dict, chunked_outputs, chunk_sizes): """ Aggregate outputs from microbatches into outputs for the whole batch. Scalar values are aggregated by `mean`, array values are concatenated along the first (batch) axis. """ - result = {} + result = OrderedDict() for output_name in outputs_dict: chunked_output = [chunk_outputs[output_name] for chunk_outputs in chunked_outputs]