diff --git a/MindElec/examples/data_driven/parameterization/src/dataset.py b/MindElec/examples/data_driven/parameterization/src/dataset.py index 00f41cd06a8d796728fe36a1c882ba6f01e68137..026d5c8a6e26422647e078937ea7f4fa7bd998a3 100755 --- a/MindElec/examples/data_driven/parameterization/src/dataset.py +++ b/MindElec/examples/data_driven/parameterization/src/dataset.py @@ -43,6 +43,7 @@ def create_dataset(opt): """ load data """ + data_input_path = opt.input_path data_label_path = opt.label_path diff --git a/MindElec/examples/data_driven/pointcloud/S_parameter/src/dataset.py b/MindElec/examples/data_driven/pointcloud/S_parameter/src/dataset.py index 21f06a6f7e72081bae861281538f1001436428d3..4f464a041d5bcf9eaf51dd57790f1d6549425c0e 100644 --- a/MindElec/examples/data_driven/pointcloud/S_parameter/src/dataset.py +++ b/MindElec/examples/data_driven/pointcloud/S_parameter/src/dataset.py @@ -26,6 +26,7 @@ SAVE_DATA_PATH = "./" def custom_normalize(dataset, mean=None, std=None): """ custom normalization """ + ori_shape = dataset.shape dataset = dataset.reshape(ori_shape[0], -1) dataset = np.transpose(dataset) diff --git a/MindElec/examples/data_driven/pointcloud/S_parameter/train.py b/MindElec/examples/data_driven/pointcloud/S_parameter/train.py index 4930b47556f954d6e02ea7b433571ce026ea4b5a..058706e3be26df681c2d88e4c649f92a5ab381c7 100644 --- a/MindElec/examples/data_driven/pointcloud/S_parameter/train.py +++ b/MindElec/examples/data_driven/pointcloud/S_parameter/train.py @@ -48,6 +48,7 @@ opt = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, save_graphs=False, device_target="Ascend", device_id=opt.device_num) def init_weight(net): + """init_weight""" for _, cell in net.cells_and_names(): if isinstance(cell, (nn.Conv3d, nn.Dense)): diff --git a/MindElec/examples/data_driven/pointcloud/data_compression/src/dataset.py b/MindElec/examples/data_driven/pointcloud/data_compression/src/dataset.py index 81f68e677a2c740775824dfdcf3b7dd14ed0095a..13a32336e0147828055cecd857544ed70124e4cc 100644 --- a/MindElec/examples/data_driven/pointcloud/data_compression/src/dataset.py +++ b/MindElec/examples/data_driven/pointcloud/data_compression/src/dataset.py @@ -31,6 +31,7 @@ SAVE_DATA_PATH = "./" def generate_data(input_path): """generate training data and data configuration""" + space_temp = np.load(input_path) print("data load finish") diff --git a/MindElec/examples/data_driven/pointcloud/full_em/src/dataset.py b/MindElec/examples/data_driven/pointcloud/full_em/src/dataset.py index 0f4da3370b4ee3b79fa3dfb1e938a436d05ec1e4..6c10c01eae04a498540f55437b1d2c2edd74720f 100644 --- a/MindElec/examples/data_driven/pointcloud/full_em/src/dataset.py +++ b/MindElec/examples/data_driven/pointcloud/full_em/src/dataset.py @@ -21,6 +21,7 @@ from mindelec.data import Dataset, ExistedDataConfig def create_dataset(data_path, batch_size=8, shuffle=True, drop_remainder=True, is_train=True): """create dataset""" + input_path = data_path + "inputs.npy" label_path = data_path + "label.npy" electromagnetic = ExistedDataConfig(name="electromagnetic",