diff --git a/nnfwtbn/model.py b/nnfwtbn/model.py index ee6c55ddf96fce40a78bfe4abdc20f1113ac2895..22708267bc4b8cdda12ff5cfa545844252352660 100644 --- a/nnfwtbn/model.py +++ b/nnfwtbn/model.py @@ -669,6 +669,7 @@ class HepNet: as variable_file: scales = self.norms[fold_i].scales offsets = self.norms[fold_i].offsets + offsets = [o / s for o, s in zip(offsets, scales)] inputs = [dict(name=v, offset=o, scale=s) for v, o, s in zip(self.input_list, offsets, scales)]