修复预测脚本和训练脚本的执行bug
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user