diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py index 007fec80964e300315c59f3d7fa4166b9d10fa70..5672c3f9a5c58f48fa42c7fe6f357f8954665f24 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py @@ -168,7 +168,8 @@ class TensorDataProcessor(PytorchDataProcessor): def _analyze_tensor(self, tensor, suffix): dump_data_name, file_path = self.get_save_file_path(suffix) if not path_len_exceeds_limit(file_path): - torch.save(tensor, file_path) + saved_tensor = tensor.contiguous().detach() + torch.save(saved_tensor, file_path) change_mode(file_path, FileCheckConst.DATA_FILE_AUTHORITY) else: logger.warning(f'The file path {file_path} length exceeds limit.')