修复预测脚本和训练脚本的执行bug

This commit is contained in:
2025-10-21 19:49:21 +08:00
parent b9cce1d733
commit 4fb2da1366
10 changed files with 87 additions and 45 deletions

View File

@@ -20,10 +20,10 @@ class Normalizer:
self.params['max_abs'] = data.abs().max(dim=0)[0]
return self
def load_params(self,method = "minmax",min_in = 0,max_in = 0,mean_in =0,std=0,max_abs=0):
def load_params(self,method = "minmax",min_in = [],max_in = [],mean_in =[],std=[],max_abs=[]):
self.method = method
self.params['min'] = min_in
self.params['max'] = max_in
self.params['min'] = torch.tensor(min_in)
self.params['max'] = torch.tensor(max_in)
self.params['mean'] = mean_in
self.params['std'] = std
self.params['max_abs'] = max_abs