Skip to content

Commit 334f04c

Browse files
committed
add some comments for pbm_norm config
1 parent fb16bf6 commit 334f04c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

torchhydro/configs/config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,7 @@ def default_config_file():
204204
"ssma",
205205
"susma",
206206
],
207+
# NOTE: pbm_norm is True means norm and denorm for differentiable models; if you use pure data-driven models, you should set it as False
207208
"pbm_norm": False,
208209
},
209210
"stat_dict_file": None,

torchhydro/datasets/data_scalers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,7 @@ def inverse_transform(self, target_values):
320320
stat_dict = self.stat_dict
321321
target_cols = self.data_cfgs["target_cols"]
322322
if self.pbm_norm:
323-
# for pbm's output, its unit is mm/day, so we don't need to recover its unit
323+
# for (differentiable models) pbm's output, its unit is mm/day, so we don't need to recover its unit
324324
pred = target_values
325325
else:
326326
pred = _trans_norm(

0 commit comments

Comments
 (0)