Skip to content

Commit 517927a

Browse files
authored
Remove cleaning stream name when logging loss (ecmwf#763)
1 parent 7bb80eb commit 517927a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/weathergen/utils/train_logger.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def add_train(
120120
log_vals += [lr]
121121

122122
for st in self.cf.streams:
123-
st_name = _clean_name(st["name"])
123+
st_name = st["name"]
124124
loss = losses_all[st_name]
125125
stddev = stddev_all[st_name]
126126

@@ -166,7 +166,7 @@ def add_val(
166166
log_vals += [samples]
167167

168168
for st in self.cf.streams:
169-
st_name = _clean_name(st["name"])
169+
st_name = st["name"]
170170
loss = losses_all[st_name]
171171
stddev = stddev_all[st_name]
172172
for j, (lf_name, _) in enumerate(self.cf.loss_fcts_val):

0 commit comments

Comments
 (0)