| @@ -60,5 +60,5 @@ def test_SoftmaxCrossEntropyExpand(): | |||||
| loss = nn.SoftmaxCrossEntropyExpand() | loss = nn.SoftmaxCrossEntropyExpand() | ||||
| logits = Tensor(np.random.randint(0, 9, [100, 10]).astype(np.float32)) | logits = Tensor(np.random.randint(0, 9, [100, 10]).astype(np.float32)) | ||||
| labels = Tensor(np.random.randint(0, 9, [10,]).astype(np.float32)) | |||||
| labels = Tensor(np.random.randint(0, 9, [10, ]).astype(np.float32)) | |||||
| _executor.compile(loss, logits, labels) | _executor.compile(loss, logits, labels) | ||||
| @@ -17,7 +17,9 @@ import numpy as np | |||||
| import os | import os | ||||
| import pytest | import pytest | ||||
| import stat | import stat | ||||
| import time | |||||
| import mindspore.common.dtype as mstype | |||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| from mindspore import context | from mindspore import context | ||||
| from mindspore.common.parameter import Parameter | from mindspore.common.parameter import Parameter | ||||
| @@ -17,6 +17,7 @@ import numpy as np | |||||
| from mobilenetv2_combined import MobileNetV2 | from mobilenetv2_combined import MobileNetV2 | ||||
| import mindspore.context as context | import mindspore.context as context | ||||
| import mindspore.ops.operations as P | |||||
| from mindspore import Tensor | from mindspore import Tensor | ||||
| from mindspore import nn | from mindspore import nn | ||||
| from mindspore.nn.layer import combined | from mindspore.nn.layer import combined | ||||
| @@ -14,8 +14,8 @@ | |||||
| # ============================================================================ | # ============================================================================ | ||||
| """ test_graph_summary """ | """ test_graph_summary """ | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import numpy as np | |||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| from mindspore import Model, context | from mindspore import Model, context | ||||
| @@ -15,9 +15,9 @@ | |||||
| """Test histogram summary.""" | """Test histogram summary.""" | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import tempfile | import tempfile | ||||
| import numpy as np | |||||
| from mindspore.common.tensor import Tensor | from mindspore.common.tensor import Tensor | ||||
| from mindspore.train.summary._summary_adapter import _calc_histogram_bins | from mindspore.train.summary._summary_adapter import _calc_histogram_bins | ||||
| @@ -19,8 +19,8 @@ | |||||
| @Desc : test summary function | @Desc : test summary function | ||||
| """ | """ | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import numpy as np | |||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| from mindspore import Model, context | from mindspore import Model, context | ||||
| @@ -19,10 +19,11 @@ | |||||
| @Desc : test summary function | @Desc : test summary function | ||||
| """ | """ | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import pytest | |||||
| import random | import random | ||||
| import numpy as np | |||||
| import pytest | |||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| from mindspore.common.tensor import Tensor | from mindspore.common.tensor import Tensor | ||||
| @@ -19,8 +19,8 @@ | |||||
| @Desc : test summary function of abnormal input | @Desc : test summary function of abnormal input | ||||
| """ | """ | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import numpy as np | |||||
| from mindspore.common.tensor import Tensor | from mindspore.common.tensor import Tensor | ||||
| from mindspore.train.summary.summary_record import SummaryRecord | from mindspore.train.summary.summary_record import SummaryRecord | ||||
| @@ -19,8 +19,8 @@ | |||||
| @Desc : test summary function | @Desc : test summary function | ||||
| """ | """ | ||||
| import logging | import logging | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import numpy as np | |||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| from mindspore.common.tensor import Tensor | from mindspore.common.tensor import Tensor | ||||
| @@ -13,10 +13,10 @@ | |||||
| # limitations under the License. | # limitations under the License. | ||||
| # ============================================================================ | # ============================================================================ | ||||
| """test callback function.""" | """test callback function.""" | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import pytest | |||||
| import stat | import stat | ||||
| import numpy as np | |||||
| import pytest | |||||
| import mindspore.common.dtype as mstype | import mindspore.common.dtype as mstype | ||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||
| @@ -14,9 +14,9 @@ | |||||
| # ============================================================================ | # ============================================================================ | ||||
| """ test_initializer """ | """ test_initializer """ | ||||
| import math | import math | ||||
| from functools import reduce | |||||
| import numpy as np | import numpy as np | ||||
| import pytest as py | import pytest as py | ||||
| from functools import reduce | |||||
| from scipy import stats | from scipy import stats | ||||
| import mindspore as ms | import mindspore as ms | ||||
| @@ -13,11 +13,11 @@ | |||||
| # limitations under the License. | # limitations under the License. | ||||
| # ============================================================================ | # ============================================================================ | ||||
| """ut for model serialize(save/load)""" | """ut for model serialize(save/load)""" | ||||
| import numpy as np | |||||
| import os | import os | ||||
| import pytest | |||||
| import stat | import stat | ||||
| import time | import time | ||||
| import pytest | |||||
| import numpy as np | |||||
| import mindspore.common.dtype as mstype | import mindspore.common.dtype as mstype | ||||
| import mindspore.nn as nn | import mindspore.nn as nn | ||||