def load_weight(self):
load_path = self.program_info['path_save'] + self.program_info[
'save_name']
if os.path.isfile(load_path):
self.weight, self.tensor_info, self.update_info, self.tensor_shape = \
BasicFunctions_szz.load_pr(load_path, ['weight', 'tensor_info', 'update_info', 'tensor_shape'])
self.weight.data = self.weight.data.to(self.device)
def load_accuracy(self):
load_path = self.program_info['path_save'] + self.program_info[
'save_name']
if os.path.isfile(load_path):
self.accuracy, self.test_info = BasicFunctions_szz.load_pr(
load_path, ['accuracy', 'test_info'])
def load_gtn(self):
load_path = self.program_info['path_save'] + self.program_info[
'save_name']
if os.path.isfile(load_path):
self.tensor_data, self.tensor_info, self.update_info = \
BasicFunctions_szz.load_pr(load_path, ['tensor_data', 'tensor_info', 'update_info'])