mindspore 2.1.0__cp38-cp38-manylinux1_x86_64.whl → 2.2.11__cp38-cp38-manylinux1_x86_64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of mindspore might be problematic. Click here for more details.
- mindspore/.commit_id +1 -1
- mindspore/__init__.py +4 -1
- mindspore/_akg/akg/build_module.py +5 -6
- mindspore/_akg/akg/composite/build_module.py +139 -22
- mindspore/_akg/akg/composite/split_stitch.py +10 -11
- mindspore/_akg/akg/ms/info_version_adapt.py +67 -1
- mindspore/_akg/akg/tvm/api.py +4 -3
- mindspore/_akg/akg/tvm/autotvm/__init__.py +1 -2
- mindspore/_akg/akg/tvm/autotvm/graph_tuner/base_graph_tuner.py +1 -5
- mindspore/_akg/akg/tvm/autotvm/measure/__init__.py +1 -1
- mindspore/_akg/akg/tvm/autotvm/measure/measure.py +1 -10
- mindspore/_akg/akg/tvm/autotvm/measure/measure_methods.py +1 -372
- mindspore/_akg/akg/tvm/build_module.py +16 -1
- mindspore/_akg/akg/tvm/contrib/graph_runtime.py +0 -53
- mindspore/_akg/akg/tvm/hybrid/parser.py +7 -6
- mindspore/_akg/akg/tvm/ir_builder.py +1 -1
- mindspore/_akg/akg/tvm/module.py +1 -2
- mindspore/_akg/akg/tvm/stmt.py +2 -2
- mindspore/_akg/akg/utils/ascend_profilier/cann_file_parser.py +76 -0
- mindspore/_akg/akg/utils/ascend_profilier/file_manager.py +56 -0
- mindspore/_akg/akg/utils/ascend_profilier/op_summary_bean.py +23 -0
- mindspore/_akg/akg/utils/ascend_profilier/op_summary_headers.py +8 -0
- mindspore/_akg/akg/utils/ascend_profilier/op_summary_parser.py +42 -0
- mindspore/_akg/akg/utils/ascend_profilier/path_manager.py +65 -0
- mindspore/_akg/akg/utils/composite_op_helper.py +16 -12
- mindspore/_akg/akg/utils/dump_ascend_meta.py +22 -3
- mindspore/_akg/akg/utils/kernel_exec.py +98 -274
- mindspore/_akg/akg/utils/result_analysis.py +4 -24
- mindspore/_akg/akg/utils/tbe_codegen_utils.py +219 -0
- mindspore/_akg/akg/utils/util.py +56 -1
- mindspore/_c_dataengine.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_check_jit_forbidden_api.py +3 -1
- mindspore/_checkparam.py +23 -29
- mindspore/_extends/graph_kernel/__init__.py +0 -1
- mindspore/_extends/graph_kernel/model/graph_split.py +84 -76
- mindspore/_extends/graph_kernel/model/model_builder.py +9 -50
- mindspore/_extends/graph_kernel/splitter.py +4 -11
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +122 -15
- mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +84 -67
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +4 -2
- mindspore/_extends/parallel_compile/akg_compiler/util.py +10 -7
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +2 -2
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +6 -5
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job.py +1 -1
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -1
- mindspore/_extends/parse/__init__.py +13 -15
- mindspore/_extends/parse/namespace.py +7 -33
- mindspore/_extends/parse/parser.py +67 -72
- mindspore/_extends/parse/resources.py +1 -1
- mindspore/_extends/parse/standard_method.py +86 -106
- mindspore/_extends/parse/trope.py +1 -1
- mindspore/_extends/remote/kernel_build_server.py +25 -7
- mindspore/_extends/remote/kernel_build_server_akg_v2.py +55 -0
- mindspore/_install_custom.py +43 -0
- mindspore/_mindspore_offline_debug.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/amp.py +47 -11
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- mindspore/boost/boost.py +1 -8
- mindspore/boost/boost_cell_wrapper.py +3 -2
- mindspore/boost/grad_accumulation.py +1 -1
- mindspore/boost/group_loss_scale_manager.py +8 -7
- mindspore/common/__init__.py +5 -3
- mindspore/common/_jit_fallback_utils.py +6 -0
- mindspore/common/_register_for_adapter.py +2 -0
- mindspore/common/_register_for_tensor.py +2 -2
- mindspore/common/_stub_tensor.py +13 -0
- mindspore/common/_utils.py +29 -0
- mindspore/common/api.py +174 -259
- mindspore/common/auto_dynamic_shape.py +494 -0
- mindspore/common/dtype.py +18 -11
- mindspore/common/dump.py +6 -4
- mindspore/common/initializer.py +14 -14
- mindspore/common/jit_config.py +33 -15
- mindspore/common/lazy_inline.py +126 -7
- mindspore/common/mindir_util.py +101 -0
- mindspore/common/parameter.py +51 -41
- mindspore/common/seed.py +4 -4
- mindspore/common/sparse_tensor.py +13 -14
- mindspore/common/tensor.py +243 -165
- mindspore/communication/__init__.py +7 -4
- mindspore/communication/_comm_helper.py +83 -4
- mindspore/communication/management.py +152 -84
- mindspore/config/op_info.config +14 -3
- mindspore/config/super_bar_config.json +4 -2
- mindspore/context.py +152 -61
- mindspore/dataset/__init__.py +5 -5
- mindspore/dataset/audio/__init__.py +2 -2
- mindspore/dataset/audio/transforms.py +52 -52
- mindspore/dataset/callback/ds_callback.py +16 -2
- mindspore/dataset/core/config.py +68 -51
- mindspore/dataset/engine/cache_client.py +33 -7
- mindspore/dataset/engine/datasets.py +250 -112
- mindspore/dataset/engine/datasets_audio.py +43 -211
- mindspore/dataset/engine/datasets_standard_format.py +16 -35
- mindspore/dataset/engine/datasets_text.py +43 -67
- mindspore/dataset/engine/datasets_user_defined.py +86 -100
- mindspore/dataset/engine/datasets_vision.py +219 -1029
- mindspore/dataset/engine/iterators.py +11 -4
- mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +4 -0
- mindspore/dataset/engine/obs/util.py +3 -0
- mindspore/dataset/engine/samplers.py +1 -1
- mindspore/dataset/engine/validators.py +19 -5
- mindspore/dataset/text/__init__.py +3 -3
- mindspore/dataset/text/transforms.py +101 -127
- mindspore/dataset/text/utils.py +205 -138
- mindspore/dataset/transforms/__init__.py +1 -1
- mindspore/dataset/transforms/py_transforms_util.py +40 -12
- mindspore/dataset/transforms/transforms.py +95 -40
- mindspore/dataset/utils/browse_dataset.py +8 -2
- mindspore/dataset/utils/line_reader.py +17 -19
- mindspore/dataset/vision/__init__.py +3 -3
- mindspore/dataset/vision/c_transforms.py +6 -3
- mindspore/dataset/vision/transforms.py +409 -287
- mindspore/dataset/vision/utils.py +13 -14
- mindspore/dataset/vision/validators.py +11 -1
- mindspore/experimental/map_parameter.py +14 -0
- mindspore/{nn/optim_ex → experimental/optim}/__init__.py +30 -29
- mindspore/{nn/optim_ex → experimental/optim}/adam.py +60 -67
- mindspore/{nn/optim_ex → experimental/optim}/adamw.py +181 -203
- mindspore/experimental/optim/lr_scheduler.py +1427 -0
- mindspore/{nn/optim_ex → experimental/optim}/optimizer.py +252 -259
- mindspore/{nn/optim_ex → experimental/optim}/sgd.py +147 -152
- mindspore/gen_ops.py +273 -0
- mindspore/include/OWNERS +0 -1
- mindspore/include/api/data_type.h +2 -1
- mindspore/include/api/graph.h +0 -15
- mindspore/include/api/kernel.h +2 -0
- mindspore/include/api/kernel_api.h +37 -12
- mindspore/include/api/model.h +17 -14
- mindspore/include/api/status.h +8 -3
- mindspore/include/api/types.h +37 -4
- mindspore/include/c_api/ms/abstract.h +67 -0
- mindspore/include/c_api/ms/attribute.h +197 -0
- mindspore/include/c_api/ms/base/handle_types.h +43 -0
- mindspore/include/c_api/ms/base/macros.h +32 -0
- mindspore/include/c_api/ms/base/status.h +33 -0
- mindspore/include/c_api/ms/base/types.h +282 -0
- mindspore/include/c_api/ms/context.h +102 -0
- mindspore/include/c_api/ms/graph.h +160 -0
- mindspore/include/c_api/ms/node.h +606 -0
- mindspore/include/c_api/ms/tensor.h +161 -0
- mindspore/include/c_api/ms/value.h +84 -0
- mindspore/include/dataset/constants.h +6 -5
- mindspore/include/dataset/execute.h +23 -13
- mindspore/include/dataset/text.h +26 -26
- mindspore/include/dataset/transforms.h +13 -13
- mindspore/include/dataset/vision.h +60 -60
- mindspore/include/dataset/vision_ascend.h +5 -6
- mindspore/include/dataset/vision_lite.h +17 -17
- mindspore/include/mindapi/base/type_id.h +1 -0
- mindspore/include/mindapi/base/types.h +1 -0
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libjemalloc.so.2 +0 -0
- mindspore/lib/libmindspore.so +0 -0
- mindspore/lib/libmindspore_backend.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_glog.so.0 +0 -0
- mindspore/lib/libmindspore_gpr.so.15 +0 -0
- mindspore/lib/libmindspore_grpc++.so.1 +0 -0
- mindspore/lib/libmindspore_grpc.so.15 +0 -0
- mindspore/lib/libmindspore_shared_lib.so +0 -0
- mindspore/lib/libnnacl.so +0 -0
- mindspore/lib/libopencv_core.so.4.5 +0 -0
- mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
- mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
- mindspore/lib/libps_cache.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend310/aic-ascend310-ops-info.json +123 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend310p/aic-ascend310p-ops-info.json +123 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json +158 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910b/aic-ascend910b-ops-info.json +37 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/add_dsl.py +46 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/add_tik.py +51 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/kv_cache_mgr.py +241 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/matmul_tik.py +212 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/add_dsl.py +46 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/add_tik.py +51 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/kv_cache_mgr.py +241 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/matmul_tik.py +212 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_proto/libop_proto.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/config/cust_aicpu_kernel.json +8998 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
- mindspore/lib/plugin/ascend/libakg.so +0 -0
- mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
- mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
- mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
- mindspore/lib/plugin/cpu/libakg.so +0 -0
- mindspore/lib/plugin/gpu/libcuda_ops.so.10 +0 -0
- mindspore/lib/plugin/gpu/libcuda_ops.so.11 +0 -0
- mindspore/lib/plugin/gpu10.1/libakg.so +0 -0
- mindspore/lib/plugin/gpu10.1/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.1/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.1/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.6/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.6/libnccl.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.1 +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.10.1 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.1 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.6 +0 -0
- mindspore/mindrecord/tools/imagenet_to_mr.py +1 -1
- mindspore/mindrecord/tools/mnist_to_mr.py +2 -2
- mindspore/nn/__init__.py +0 -2
- mindspore/nn/cell.py +313 -74
- mindspore/nn/dynamic_lr.py +21 -21
- mindspore/nn/layer/activation.py +22 -30
- mindspore/nn/layer/basic.py +15 -13
- mindspore/nn/layer/channel_shuffle.py +1 -1
- mindspore/nn/layer/container.py +271 -9
- mindspore/nn/layer/conv.py +323 -204
- mindspore/nn/layer/dense.py +8 -5
- mindspore/nn/layer/embedding.py +33 -27
- mindspore/nn/layer/flash_attention.py +61 -95
- mindspore/nn/layer/image.py +8 -6
- mindspore/nn/layer/math.py +16 -25
- mindspore/nn/layer/normalization.py +107 -66
- mindspore/nn/layer/padding.py +1 -1
- mindspore/nn/layer/pooling.py +131 -109
- mindspore/nn/layer/rnn_cells.py +27 -22
- mindspore/nn/layer/rnns.py +13 -16
- mindspore/nn/layer/thor_layer.py +1 -1
- mindspore/nn/layer/transformer.py +221 -154
- mindspore/nn/learning_rate_schedule.py +9 -1
- mindspore/nn/loss/loss.py +235 -174
- mindspore/nn/optim/ada_grad.py +2 -1
- mindspore/nn/optim/adadelta.py +1 -0
- mindspore/nn/optim/adafactor.py +2 -1
- mindspore/nn/optim/adam.py +7 -4
- mindspore/nn/optim/adamax.py +3 -2
- mindspore/nn/optim/adasum.py +2 -2
- mindspore/nn/optim/asgd.py +2 -3
- mindspore/nn/optim/ftrl.py +6 -5
- mindspore/nn/optim/lamb.py +7 -4
- mindspore/nn/optim/lars.py +1 -1
- mindspore/nn/optim/lazyadam.py +5 -3
- mindspore/nn/optim/momentum.py +2 -1
- mindspore/nn/optim/optimizer.py +53 -4
- mindspore/nn/optim/proximal_ada_grad.py +3 -4
- mindspore/nn/optim/rmsprop.py +4 -3
- mindspore/nn/optim/rprop.py +23 -12
- mindspore/nn/optim/sgd.py +26 -11
- mindspore/nn/optim/thor.py +9 -7
- mindspore/nn/probability/bijector/bijector.py +5 -5
- mindspore/nn/probability/bijector/power_transform.py +27 -27
- mindspore/nn/probability/bijector/softplus.py +3 -3
- mindspore/nn/probability/distribution/_utils/custom_ops.py +3 -3
- mindspore/nn/probability/distribution/bernoulli.py +5 -5
- mindspore/nn/probability/distribution/beta.py +3 -3
- mindspore/nn/probability/distribution/categorical.py +7 -7
- mindspore/nn/probability/distribution/cauchy.py +0 -1
- mindspore/nn/probability/distribution/distribution.py +3 -3
- mindspore/nn/probability/distribution/gamma.py +3 -3
- mindspore/nn/probability/distribution/geometric.py +4 -4
- mindspore/nn/probability/distribution/gumbel.py +4 -4
- mindspore/nn/probability/distribution/log_normal.py +2 -2
- mindspore/nn/probability/distribution/logistic.py +2 -2
- mindspore/nn/probability/distribution/poisson.py +4 -4
- mindspore/nn/probability/distribution/transformed_distribution.py +3 -3
- mindspore/nn/probability/distribution/uniform.py +6 -6
- mindspore/nn/wrap/__init__.py +4 -2
- mindspore/nn/wrap/cell_wrapper.py +87 -34
- mindspore/nn/wrap/grad_reducer.py +8 -5
- mindspore/nn/wrap/loss_scale.py +105 -42
- mindspore/numpy/array_creations.py +1 -2
- mindspore/numpy/array_ops.py +3 -2
- mindspore/numpy/utils_const.py +5 -5
- mindspore/offline_debug/convert_async.py +2 -2
- mindspore/ops/_grad_experimental/__init__.py +0 -5
- mindspore/ops/_grad_experimental/grad_array_ops.py +2 -3
- mindspore/ops/_grad_experimental/grad_comm_ops.py +15 -2
- mindspore/ops/_grad_experimental/grad_debug_ops.py +0 -37
- mindspore/ops/_grad_experimental/grad_implementations.py +11 -1
- mindspore/ops/_grad_experimental/grad_inner_ops.py +2 -216
- mindspore/ops/_grad_experimental/grad_math_ops.py +19 -199
- mindspore/ops/_grad_experimental/grad_sparse.py +15 -0
- mindspore/ops/_grad_experimental/grad_sparse_ops.py +3 -3
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -1
- mindspore/ops/_op_impl/aicpu/__init__.py +14 -2
- mindspore/ops/_op_impl/aicpu/add.py +3 -3
- mindspore/ops/_op_impl/aicpu/bias_add_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/count_nonzero.py +43 -0
- mindspore/ops/_op_impl/{_custom_op/flash_attention/constants.py → aicpu/eps.py} +18 -27
- mindspore/ops/_op_impl/aicpu/gamma.py +2 -2
- mindspore/ops/_op_impl/aicpu/linear_sum_assignment.py +21 -2
- mindspore/ops/_op_impl/aicpu/log_uniform_candidate_sampler.py +6 -3
- mindspore/ops/_op_impl/aicpu/lu_unpack_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/multinomial.py +3 -3
- mindspore/ops/_op_impl/aicpu/parameterized_truncated_normal.py +15 -7
- mindspore/ops/_op_impl/aicpu/random_categorical.py +39 -19
- mindspore/ops/_op_impl/aicpu/random_choice_with_mask.py +5 -2
- mindspore/ops/_op_impl/aicpu/random_poisson.py +103 -52
- mindspore/ops/_op_impl/aicpu/random_shuffle.py +17 -15
- mindspore/ops/_op_impl/aicpu/{sparseaddmm.py → sparse_addmm.py} +2 -2
- mindspore/ops/_op_impl/aicpu/{sparsesparsemaximum.py → sparse_sparse_maximum.py} +4 -4
- mindspore/ops/_op_impl/aicpu/standard_laplace.py +5 -5
- mindspore/ops/_op_impl/aicpu/standard_normal.py +5 -5
- mindspore/ops/_op_impl/aicpu/truncated_normal.py +9 -7
- mindspore/ops/_op_impl/aicpu/uniform.py +5 -3
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +8 -4
- mindspore/ops/_op_impl/aicpu/uniform_int.py +5 -5
- mindspore/ops/_op_impl/aicpu/uniform_real.py +4 -4
- mindspore/ops/_op_impl/tbe/__init__.py +4 -4
- mindspore/ops/_op_impl/tbe/inplace_index_add.py +7 -3
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +2 -0
- mindspore/ops/_primitive_cache.py +1 -1
- mindspore/ops/_tracefunc.py +45 -13
- mindspore/ops/_utils/utils.py +6 -1
- mindspore/ops/_vmap/vmap_array_ops.py +3 -3
- mindspore/ops/_vmap/vmap_base.py +3 -3
- mindspore/ops/_vmap/vmap_convolution_ops.py +1 -1
- mindspore/ops/_vmap/vmap_grad_math_ops.py +6 -4
- mindspore/ops/_vmap/vmap_math_ops.py +5 -2
- mindspore/ops/_vmap/vmap_nn_ops.py +61 -7
- mindspore/ops/arg_dtype_cast.py +54 -0
- mindspore/ops/composite/base.py +37 -10
- mindspore/ops/composite/math_ops.py +5 -4
- mindspore/ops/composite/multitype_ops/_compile_utils.py +275 -73
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +16 -9
- mindspore/ops/composite/multitype_ops/add_impl.py +43 -4
- mindspore/ops/composite/multitype_ops/getitem_impl.py +42 -4
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +6 -0
- mindspore/ops/composite/multitype_ops/setitem_impl.py +2 -1
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +9 -0
- mindspore/ops/deprecated.py +304 -0
- mindspore/ops/function/__init__.py +4 -1
- mindspore/ops/function/array_func.py +174 -193
- mindspore/ops/function/clip_func.py +81 -13
- mindspore/ops/function/debug_func.py +1 -1
- mindspore/ops/function/grad/grad_func.py +18 -9
- mindspore/ops/function/image_func.py +10 -4
- mindspore/ops/function/linalg_func.py +5 -5
- mindspore/ops/function/math_func.py +575 -386
- mindspore/ops/function/nn_func.py +568 -260
- mindspore/ops/function/random_func.py +88 -57
- mindspore/ops/function/sparse_func.py +1 -1
- mindspore/ops/function/sparse_unary_func.py +14 -12
- mindspore/ops/function/vmap_func.py +6 -5
- mindspore/ops/functional.py +15 -10
- mindspore/ops/op_info_register.py +244 -25
- mindspore/ops/operations/__init__.py +31 -19
- mindspore/ops/operations/_grad_ops.py +71 -7
- mindspore/ops/operations/_inner_ops.py +350 -17
- mindspore/ops/operations/_quant_ops.py +4 -8
- mindspore/ops/operations/_sequence_ops.py +42 -0
- mindspore/ops/operations/array_ops.py +68 -282
- mindspore/ops/operations/comm_ops.py +107 -59
- mindspore/ops/operations/custom_ops.py +94 -70
- mindspore/ops/operations/debug_ops.py +8 -4
- mindspore/ops/operations/image_ops.py +18 -12
- mindspore/ops/operations/inner_ops.py +26 -3
- mindspore/ops/operations/math_ops.py +192 -144
- mindspore/ops/operations/nn_ops.py +857 -489
- mindspore/ops/operations/other_ops.py +0 -22
- mindspore/ops/operations/random_ops.py +53 -111
- mindspore/ops/operations/sparse_ops.py +3 -1
- mindspore/ops/primitive.py +24 -18
- mindspore/parallel/_auto_parallel_context.py +68 -8
- mindspore/parallel/_cost_model_context.py +2 -2
- mindspore/parallel/_offload_context.py +17 -3
- mindspore/parallel/_parallel_serialization.py +12 -5
- mindspore/parallel/_ps_context.py +12 -0
- mindspore/parallel/_tensor.py +18 -13
- mindspore/parallel/_transformer/layers.py +5 -3
- mindspore/parallel/_transformer/loss.py +1 -0
- mindspore/parallel/_transformer/moe.py +2 -2
- mindspore/parallel/_transformer/op_parallel_config.py +12 -1
- mindspore/parallel/_transformer/transformer.py +23 -3
- mindspore/parallel/_utils.py +11 -7
- mindspore/parallel/algo_parameter_config.py +85 -5
- mindspore/parallel/checkpoint_transform.py +19 -12
- mindspore/parallel/shard.py +21 -14
- mindspore/profiler/common/struct_type.py +3 -3
- mindspore/profiler/common/util.py +4 -2
- mindspore/profiler/envprofiling.py +1 -1
- mindspore/profiler/parser/aicpu_data_parser.py +5 -3
- mindspore/profiler/parser/ascend_flops_generator.py +2 -2
- mindspore/profiler/parser/ascend_fpbp_generator.py +1 -1
- mindspore/profiler/parser/ascend_hccl_generator.py +249 -12
- mindspore/profiler/parser/ascend_msprof_exporter.py +150 -255
- mindspore/profiler/parser/ascend_msprof_generator.py +204 -17
- mindspore/profiler/parser/ascend_op_generator.py +6 -6
- mindspore/profiler/parser/ascend_steptrace_generator.py +6 -4
- mindspore/profiler/parser/ascend_timeline_generator.py +14 -187
- mindspore/profiler/parser/base_timeline_generator.py +10 -8
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +16 -12
- mindspore/profiler/parser/flops_parser.py +15 -11
- mindspore/profiler/parser/framework_parser.py +38 -22
- mindspore/profiler/parser/hccl_parser.py +16 -12
- mindspore/profiler/parser/integrator.py +22 -11
- mindspore/profiler/parser/memory_usage_parser.py +2 -2
- mindspore/profiler/parser/minddata_analyzer.py +12 -14
- mindspore/profiler/parser/minddata_pipeline_parser.py +1 -1
- mindspore/profiler/parser/msadvisor_parser.py +8 -4
- mindspore/profiler/parser/op_intermediate_parser.py +5 -2
- mindspore/profiler/parser/optime_parser.py +1 -1
- mindspore/profiler/parser/profiler_info.py +21 -2
- mindspore/profiler/parser/step_trace_parser.py +11 -14
- mindspore/profiler/profiling.py +179 -89
- mindspore/rewrite/api/node.py +102 -19
- mindspore/rewrite/api/node_type.py +5 -1
- mindspore/rewrite/api/pattern_engine.py +1 -1
- mindspore/rewrite/api/scoped_value.py +9 -17
- mindspore/rewrite/api/symbol_tree.py +131 -47
- mindspore/rewrite/ast_helpers/__init__.py +2 -1
- mindspore/rewrite/ast_helpers/ast_finder.py +129 -0
- mindspore/rewrite/ast_helpers/ast_modifier.py +116 -104
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +93 -46
- mindspore/rewrite/common/rewrite_elog.py +5 -1
- mindspore/rewrite/namer.py +33 -24
- mindspore/rewrite/namespace.py +14 -5
- mindspore/{_extends/graph_kernel/expanders/complex → rewrite/node}/__init__.py +9 -9
- mindspore/rewrite/node/call_function.py +79 -0
- mindspore/rewrite/node/cell_container.py +135 -0
- mindspore/rewrite/node/control_flow.py +88 -0
- mindspore/rewrite/{node.py → node/node.py} +273 -234
- mindspore/rewrite/node/node_manager.py +254 -0
- mindspore/rewrite/{topological_manager.py → node/node_topological_manager.py} +13 -46
- mindspore/rewrite/parsers/arguments_parser.py +22 -21
- mindspore/rewrite/parsers/assign_parser.py +216 -221
- mindspore/rewrite/parsers/attribute_parser.py +9 -7
- mindspore/rewrite/parsers/class_def_parser.py +174 -113
- mindspore/rewrite/parsers/constant_parser.py +9 -6
- mindspore/rewrite/parsers/container_parser.py +9 -7
- mindspore/rewrite/parsers/for_parser.py +42 -21
- mindspore/rewrite/parsers/function_def_parser.py +24 -16
- mindspore/rewrite/parsers/if_parser.py +28 -24
- mindspore/rewrite/parsers/module_parser.py +196 -25
- mindspore/rewrite/{parser.py → parsers/parser.py} +4 -2
- mindspore/rewrite/{parser_register.py → parsers/parser_register.py} +1 -1
- mindspore/rewrite/parsers/return_parser.py +6 -6
- mindspore/rewrite/sparsify/sparse_transformer.py +12 -3
- mindspore/rewrite/sparsify/utils.py +1 -1
- mindspore/rewrite/symbol_tree.py +523 -578
- mindspore/rewrite/symbol_tree_builder.py +9 -193
- mindspore/rewrite/symbol_tree_dumper.py +2 -2
- mindspore/run_check/_check_version.py +6 -4
- mindspore/{ops/bprop_mindir → safeguard}/__init__.py +4 -3
- mindspore/safeguard/rewrite_obfuscation.py +541 -0
- mindspore/scipy/linalg.py +1 -1
- mindspore/scipy/ops.py +55 -5
- mindspore/scipy/optimize/__init__.py +3 -2
- mindspore/scipy/optimize/linear_sum_assignment.py +38 -33
- mindspore/scipy/optimize/minimize.py +7 -3
- mindspore/train/_utils.py +7 -3
- mindspore/train/amp.py +323 -123
- mindspore/train/anf_ir_pb2.py +14 -2
- mindspore/train/callback/_backup_and_restore.py +2 -12
- mindspore/train/callback/_callback.py +29 -4
- mindspore/train/callback/_checkpoint.py +23 -8
- mindspore/train/callback/_early_stop.py +2 -2
- mindspore/train/callback/_landscape.py +4 -4
- mindspore/train/callback/_loss_monitor.py +2 -2
- mindspore/train/callback/_on_request_exit.py +2 -2
- mindspore/train/callback/_reduce_lr_on_plateau.py +3 -4
- mindspore/train/callback/_summary_collector.py +15 -8
- mindspore/train/callback/_time_monitor.py +58 -5
- mindspore/train/data_sink.py +5 -11
- mindspore/train/dataset_helper.py +84 -57
- mindspore/train/loss_scale_manager.py +2 -2
- mindspore/train/metrics/__init__.py +3 -3
- mindspore/train/metrics/cosine_similarity.py +1 -1
- mindspore/train/metrics/hausdorff_distance.py +3 -2
- mindspore/train/metrics/mean_surface_distance.py +3 -2
- mindspore/train/metrics/metric.py +39 -19
- mindspore/train/metrics/roc.py +2 -2
- mindspore/train/metrics/root_mean_square_surface_distance.py +4 -3
- mindspore/train/mind_ir_pb2.py +85 -36
- mindspore/train/model.py +187 -47
- mindspore/train/serialization.py +487 -161
- mindspore/train/summary/_summary_adapter.py +1 -1
- mindspore/train/summary/_writer_pool.py +3 -2
- mindspore/train/summary/summary_record.py +37 -17
- mindspore/train/train_thor/convert_utils.py +3 -3
- mindspore/train/train_thor/dataset_helper.py +1 -1
- mindspore/version.py +1 -1
- {mindspore-2.1.0.dist-info → mindspore-2.2.11.dist-info}/METADATA +8 -8
- {mindspore-2.1.0.dist-info → mindspore-2.2.11.dist-info}/RECORD +488 -539
- {mindspore-2.1.0.dist-info → mindspore-2.2.11.dist-info}/entry_points.txt +0 -1
- mindspore/_akg/akg/tvm/contrib/debugger/__init__.py +0 -16
- mindspore/_akg/akg/tvm/contrib/debugger/debug_result.py +0 -274
- mindspore/_akg/akg/tvm/contrib/debugger/debug_runtime.py +0 -259
- mindspore/_akg/akg/tvm/contrib/peak.py +0 -341
- mindspore/_akg/akg/tvm/contrib/rpc.py +0 -25
- mindspore/_akg/akg/tvm/contrib/xcode.py +0 -257
- mindspore/_akg/akg/tvm/exec/__init__.py +0 -17
- mindspore/_akg/akg/tvm/exec/autotvm_log_editor.py +0 -60
- mindspore/_akg/akg/tvm/exec/measure_peak.py +0 -48
- mindspore/_akg/akg/tvm/exec/query_rpc_tracker.py +0 -48
- mindspore/_akg/akg/tvm/exec/rpc_proxy.py +0 -98
- mindspore/_akg/akg/tvm/exec/rpc_server.py +0 -88
- mindspore/_akg/akg/tvm/exec/rpc_tracker.py +0 -62
- mindspore/_akg/akg/tvm/rpc/__init__.py +0 -29
- mindspore/_akg/akg/tvm/rpc/base.py +0 -182
- mindspore/_akg/akg/tvm/rpc/client.py +0 -436
- mindspore/_akg/akg/tvm/rpc/proxy.py +0 -595
- mindspore/_akg/akg/tvm/rpc/server.py +0 -413
- mindspore/_akg/akg/tvm/rpc/tornado_util.py +0 -121
- mindspore/_akg/akg/tvm/rpc/tracker.py +0 -431
- mindspore/_extends/graph_kernel/expander.py +0 -80
- mindspore/_extends/graph_kernel/expanders/__init__.py +0 -54
- mindspore/_extends/graph_kernel/expanders/_utils.py +0 -269
- mindspore/_extends/graph_kernel/expanders/addn.py +0 -33
- mindspore/_extends/graph_kernel/expanders/batchnorm.py +0 -152
- mindspore/_extends/graph_kernel/expanders/batchnorm_grad.py +0 -105
- mindspore/_extends/graph_kernel/expanders/clip_by_norm_no_div_sum.py +0 -33
- mindspore/_extends/graph_kernel/expanders/complex/abs.py +0 -30
- mindspore/_extends/graph_kernel/expanders/complex/add.py +0 -44
- mindspore/_extends/graph_kernel/expanders/complex/div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/mul.py +0 -52
- mindspore/_extends/graph_kernel/expanders/complex/real_div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/sub.py +0 -45
- mindspore/_extends/graph_kernel/expanders/conv2d.py +0 -200
- mindspore/_extends/graph_kernel/expanders/dropout_grad.py +0 -30
- mindspore/_extends/graph_kernel/expanders/equal_count.py +0 -50
- mindspore/_extends/graph_kernel/expanders/erfc.py +0 -35
- mindspore/_extends/graph_kernel/expanders/expand_dims.py +0 -50
- mindspore/_extends/graph_kernel/expanders/fused_adam.py +0 -44
- mindspore/_extends/graph_kernel/expanders/fused_adam_weight_decay.py +0 -47
- mindspore/_extends/graph_kernel/expanders/fused_mul_add.py +0 -28
- mindspore/_extends/graph_kernel/expanders/gelu_grad.py +0 -70
- mindspore/_extends/graph_kernel/expanders/gkdropout.py +0 -40
- mindspore/_extends/graph_kernel/expanders/identity.py +0 -25
- mindspore/_extends/graph_kernel/expanders/layernorm.py +0 -93
- mindspore/_extends/graph_kernel/expanders/layernorm_grad.py +0 -113
- mindspore/_extends/graph_kernel/expanders/logsoftmax.py +0 -46
- mindspore/_extends/graph_kernel/expanders/logsoftmax_grad.py +0 -36
- mindspore/_extends/graph_kernel/expanders/matmul.py +0 -80
- mindspore/_extends/graph_kernel/expanders/maximum_grad.py +0 -59
- mindspore/_extends/graph_kernel/expanders/minimum_grad.py +0 -80
- mindspore/_extends/graph_kernel/expanders/oneslike.py +0 -26
- mindspore/_extends/graph_kernel/expanders/reduce_mean.py +0 -43
- mindspore/_extends/graph_kernel/expanders/relu_grad.py +0 -32
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits.py +0 -41
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits_grad.py +0 -35
- mindspore/_extends/graph_kernel/expanders/sigmoid_grad.py +0 -31
- mindspore/_extends/graph_kernel/expanders/slice.py +0 -35
- mindspore/_extends/graph_kernel/expanders/softmax_cross_entropy_with_logits.py +0 -42
- mindspore/_extends/graph_kernel/expanders/softmax_grad_ext.py +0 -41
- mindspore/_extends/graph_kernel/expanders/softsign.py +0 -28
- mindspore/_extends/graph_kernel/expanders/sqrt_grad.py +0 -29
- mindspore/_extends/graph_kernel/expanders/square_sum_all.py +0 -44
- mindspore/_extends/graph_kernel/expanders/square_sum_v1.py +0 -37
- mindspore/_extends/graph_kernel/expanders/squared_difference.py +0 -43
- mindspore/_extends/graph_kernel/expanders/tanh_grad.py +0 -31
- mindspore/_extends/graph_kernel/model/op_infer.py +0 -506
- mindspore/dataset/datapreprocess/__init__.py +0 -20
- mindspore/dataset/datapreprocess/preprocess_imagenet_validate_dataset.py +0 -54
- mindspore/include/api/net.h +0 -142
- mindspore/nn/lr_scheduler.py +0 -262
- mindspore/ops/_grad_experimental/grad_image_ops.py +0 -248
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +0 -181
- mindspore/ops/_grad_experimental/grad_other_ops.py +0 -72
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +0 -112
- mindspore/ops/_grad_experimental/grad_sequence_ops.py +0 -351
- mindspore/ops/_op_impl/_custom_op/flash_attention/attention.py +0 -350
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_bwd.py +0 -409
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_fwd.py +0 -578
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_impl.py +0 -199
- mindspore/ops/_op_impl/_custom_op/flash_attention/tik_ops_utils.py +0 -446
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/__init__.py +0 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/sparse_tiling.py +0 -45
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/strategy.py +0 -67
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/wukong_tiling.py +0 -62
- mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Depend_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +0 -138
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Load_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Switch_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TransShape_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Unique_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Unstack_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/generate_mindir.py +0 -114
- mindspore/rewrite/node_visitor.py +0 -44
- /mindspore/{ops/_op_impl/_custom_op/flash_attention → _akg/akg/utils/ascend_profilier}/__init__.py +0 -0
- {mindspore-2.1.0.dist-info → mindspore-2.2.11.dist-info}/WHEEL +0 -0
- {mindspore-2.1.0.dist-info → mindspore-2.2.11.dist-info}/top_level.txt +0 -0
|
@@ -99,7 +99,7 @@ class AllpassBiquad(AudioTensorOperation):
|
|
|
99
99
|
|
|
100
100
|
Tutorial Examples:
|
|
101
101
|
- `Illustration of audio transforms
|
|
102
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
102
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
103
103
|
"""
|
|
104
104
|
|
|
105
105
|
@check_allpass_biquad
|
|
@@ -162,7 +162,7 @@ class AmplitudeToDB(AudioTensorOperation):
|
|
|
162
162
|
|
|
163
163
|
Tutorial Examples:
|
|
164
164
|
- `Illustration of audio transforms
|
|
165
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
165
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
166
166
|
"""
|
|
167
167
|
|
|
168
168
|
@check_amplitude_to_db
|
|
@@ -203,7 +203,7 @@ class Angle(AudioTensorOperation):
|
|
|
203
203
|
|
|
204
204
|
Tutorial Examples:
|
|
205
205
|
- `Illustration of audio transforms
|
|
206
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
206
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
207
207
|
"""
|
|
208
208
|
|
|
209
209
|
def parse(self):
|
|
@@ -256,7 +256,7 @@ class BandBiquad(AudioTensorOperation):
|
|
|
256
256
|
|
|
257
257
|
Tutorial Examples:
|
|
258
258
|
- `Illustration of audio transforms
|
|
259
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
259
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
260
260
|
"""
|
|
261
261
|
|
|
262
262
|
@check_band_biquad
|
|
@@ -323,7 +323,7 @@ class BandpassBiquad(AudioTensorOperation):
|
|
|
323
323
|
|
|
324
324
|
Tutorial Examples:
|
|
325
325
|
- `Illustration of audio transforms
|
|
326
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
326
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
327
327
|
"""
|
|
328
328
|
|
|
329
329
|
@check_bandpass_biquad
|
|
@@ -385,7 +385,7 @@ class BandrejectBiquad(AudioTensorOperation):
|
|
|
385
385
|
|
|
386
386
|
Tutorial Examples:
|
|
387
387
|
- `Illustration of audio transforms
|
|
388
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
388
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
389
389
|
"""
|
|
390
390
|
|
|
391
391
|
@check_bandreject_biquad
|
|
@@ -445,7 +445,7 @@ class BassBiquad(AudioTensorOperation):
|
|
|
445
445
|
|
|
446
446
|
Tutorial Examples:
|
|
447
447
|
- `Illustration of audio transforms
|
|
448
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
448
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
449
449
|
"""
|
|
450
450
|
|
|
451
451
|
@check_bass_biquad
|
|
@@ -495,7 +495,7 @@ class Biquad(TensorOperation):
|
|
|
495
495
|
|
|
496
496
|
Tutorial Examples:
|
|
497
497
|
- `Illustration of audio transforms
|
|
498
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
498
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
499
499
|
"""
|
|
500
500
|
|
|
501
501
|
@check_biquad
|
|
@@ -543,7 +543,7 @@ class ComplexNorm(AudioTensorOperation):
|
|
|
543
543
|
|
|
544
544
|
Tutorial Examples:
|
|
545
545
|
- `Illustration of audio transforms
|
|
546
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
546
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
547
547
|
"""
|
|
548
548
|
|
|
549
549
|
@check_complex_norm
|
|
@@ -610,7 +610,7 @@ class ComputeDeltas(AudioTensorOperation):
|
|
|
610
610
|
|
|
611
611
|
Tutorial Examples:
|
|
612
612
|
- `Illustration of audio transforms
|
|
613
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
613
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
614
614
|
"""
|
|
615
615
|
|
|
616
616
|
@check_compute_deltas
|
|
@@ -659,7 +659,7 @@ class Contrast(AudioTensorOperation):
|
|
|
659
659
|
|
|
660
660
|
Tutorial Examples:
|
|
661
661
|
- `Illustration of audio transforms
|
|
662
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
662
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
663
663
|
"""
|
|
664
664
|
|
|
665
665
|
@check_contrast
|
|
@@ -698,7 +698,7 @@ class DBToAmplitude(AudioTensorOperation):
|
|
|
698
698
|
|
|
699
699
|
Tutorial Examples:
|
|
700
700
|
- `Illustration of audio transforms
|
|
701
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
701
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
702
702
|
"""
|
|
703
703
|
|
|
704
704
|
@check_db_to_amplitude
|
|
@@ -741,7 +741,7 @@ class DCShift(AudioTensorOperation):
|
|
|
741
741
|
|
|
742
742
|
Tutorial Examples:
|
|
743
743
|
- `Illustration of audio transforms
|
|
744
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
744
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
745
745
|
"""
|
|
746
746
|
|
|
747
747
|
@check_dc_shift
|
|
@@ -783,7 +783,7 @@ class DeemphBiquad(AudioTensorOperation):
|
|
|
783
783
|
|
|
784
784
|
Tutorial Examples:
|
|
785
785
|
- `Illustration of audio transforms
|
|
786
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
786
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
787
787
|
"""
|
|
788
788
|
|
|
789
789
|
@check_deemph_biquad
|
|
@@ -839,7 +839,7 @@ class DetectPitchFrequency(AudioTensorOperation):
|
|
|
839
839
|
|
|
840
840
|
Tutorial Examples:
|
|
841
841
|
- `Illustration of audio transforms
|
|
842
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
842
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
843
843
|
"""
|
|
844
844
|
|
|
845
845
|
@check_detect_pitch_frequency
|
|
@@ -895,7 +895,7 @@ class Dither(AudioTensorOperation):
|
|
|
895
895
|
|
|
896
896
|
Tutorial Examples:
|
|
897
897
|
- `Illustration of audio transforms
|
|
898
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
898
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
899
899
|
"""
|
|
900
900
|
|
|
901
901
|
@check_dither
|
|
@@ -943,7 +943,7 @@ class EqualizerBiquad(AudioTensorOperation):
|
|
|
943
943
|
|
|
944
944
|
Tutorial Examples:
|
|
945
945
|
- `Illustration of audio transforms
|
|
946
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
946
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
947
947
|
"""
|
|
948
948
|
|
|
949
949
|
@check_equalizer_biquad
|
|
@@ -1005,7 +1005,7 @@ class Fade(AudioTensorOperation):
|
|
|
1005
1005
|
|
|
1006
1006
|
Tutorial Examples:
|
|
1007
1007
|
- `Illustration of audio transforms
|
|
1008
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1008
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1009
1009
|
"""
|
|
1010
1010
|
|
|
1011
1011
|
@check_fade
|
|
@@ -1054,7 +1054,7 @@ class Filtfilt(AudioTensorOperation):
|
|
|
1054
1054
|
|
|
1055
1055
|
Tutorial Examples:
|
|
1056
1056
|
- `Illustration of audio transforms
|
|
1057
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1057
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1058
1058
|
"""
|
|
1059
1059
|
|
|
1060
1060
|
@check_lfilter
|
|
@@ -1128,7 +1128,7 @@ class Flanger(AudioTensorOperation):
|
|
|
1128
1128
|
|
|
1129
1129
|
Tutorial Examples:
|
|
1130
1130
|
- `Illustration of audio transforms
|
|
1131
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1131
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1132
1132
|
"""
|
|
1133
1133
|
|
|
1134
1134
|
@check_flanger
|
|
@@ -1194,7 +1194,7 @@ class FrequencyMasking(AudioTensorOperation):
|
|
|
1194
1194
|
|
|
1195
1195
|
Tutorial Examples:
|
|
1196
1196
|
- `Illustration of audio transforms
|
|
1197
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1197
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1198
1198
|
|
|
1199
1199
|
.. image:: frequency_masking_original.png
|
|
1200
1200
|
|
|
@@ -1239,7 +1239,7 @@ class Gain(AudioTensorOperation):
|
|
|
1239
1239
|
|
|
1240
1240
|
Tutorial Examples:
|
|
1241
1241
|
- `Illustration of audio transforms
|
|
1242
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1242
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1243
1243
|
"""
|
|
1244
1244
|
|
|
1245
1245
|
@check_gain
|
|
@@ -1309,7 +1309,7 @@ class GriffinLim(AudioTensorOperation):
|
|
|
1309
1309
|
|
|
1310
1310
|
Tutorial Examples:
|
|
1311
1311
|
- `Illustration of audio transforms
|
|
1312
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1312
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1313
1313
|
"""
|
|
1314
1314
|
|
|
1315
1315
|
@check_griffin_lim
|
|
@@ -1366,7 +1366,7 @@ class HighpassBiquad(AudioTensorOperation):
|
|
|
1366
1366
|
|
|
1367
1367
|
Tutorial Examples:
|
|
1368
1368
|
- `Illustration of audio transforms
|
|
1369
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1369
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1370
1370
|
"""
|
|
1371
1371
|
|
|
1372
1372
|
@check_highpass_biquad
|
|
@@ -1436,7 +1436,7 @@ class InverseMelScale(AudioTensorOperation):
|
|
|
1436
1436
|
|
|
1437
1437
|
Tutorial Examples:
|
|
1438
1438
|
- `Illustration of audio transforms
|
|
1439
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1439
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1440
1440
|
"""
|
|
1441
1441
|
|
|
1442
1442
|
@check_inverse_mel_scale
|
|
@@ -1521,7 +1521,7 @@ class InverseSpectrogram(AudioTensorOperation):
|
|
|
1521
1521
|
|
|
1522
1522
|
Tutorial Examples:
|
|
1523
1523
|
- `Illustration of audio transforms
|
|
1524
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1524
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1525
1525
|
"""
|
|
1526
1526
|
|
|
1527
1527
|
@check_inverse_spectrogram
|
|
@@ -1608,7 +1608,7 @@ class LFCC(AudioTensorOperation):
|
|
|
1608
1608
|
|
|
1609
1609
|
Tutorial Examples:
|
|
1610
1610
|
- `Illustration of audio transforms
|
|
1611
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1611
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1612
1612
|
"""
|
|
1613
1613
|
|
|
1614
1614
|
@check_lfcc
|
|
@@ -1682,7 +1682,7 @@ class LFilter(AudioTensorOperation):
|
|
|
1682
1682
|
|
|
1683
1683
|
Tutorial Examples:
|
|
1684
1684
|
- `Illustration of audio transforms
|
|
1685
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1685
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1686
1686
|
"""
|
|
1687
1687
|
|
|
1688
1688
|
@check_lfilter
|
|
@@ -1741,7 +1741,7 @@ class LowpassBiquad(AudioTensorOperation):
|
|
|
1741
1741
|
|
|
1742
1742
|
Tutorial Examples:
|
|
1743
1743
|
- `Illustration of audio transforms
|
|
1744
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1744
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1745
1745
|
"""
|
|
1746
1746
|
|
|
1747
1747
|
@check_lowpass_biquad
|
|
@@ -1757,7 +1757,7 @@ class LowpassBiquad(AudioTensorOperation):
|
|
|
1757
1757
|
|
|
1758
1758
|
class Magphase(AudioTensorOperation):
|
|
1759
1759
|
"""
|
|
1760
|
-
Separate a complex-valued spectrogram with shape (..., 2) into its magnitude and phase.
|
|
1760
|
+
Separate a complex-valued spectrogram with shape :math:`(..., 2)` into its magnitude and phase.
|
|
1761
1761
|
|
|
1762
1762
|
Args:
|
|
1763
1763
|
power (float): Power of the norm, which must be non-negative. Default: ``1.0``.
|
|
@@ -1780,7 +1780,7 @@ class Magphase(AudioTensorOperation):
|
|
|
1780
1780
|
|
|
1781
1781
|
Tutorial Examples:
|
|
1782
1782
|
- `Illustration of audio transforms
|
|
1783
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1783
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1784
1784
|
"""
|
|
1785
1785
|
|
|
1786
1786
|
@check_magphase
|
|
@@ -1822,7 +1822,7 @@ class MaskAlongAxis(AudioTensorOperation):
|
|
|
1822
1822
|
|
|
1823
1823
|
Tutorial Examples:
|
|
1824
1824
|
- `Illustration of audio transforms
|
|
1825
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1825
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1826
1826
|
"""
|
|
1827
1827
|
|
|
1828
1828
|
@check_mask_along_axis
|
|
@@ -1873,7 +1873,7 @@ class MaskAlongAxisIID(AudioTensorOperation):
|
|
|
1873
1873
|
|
|
1874
1874
|
Tutorial Examples:
|
|
1875
1875
|
- `Illustration of audio transforms
|
|
1876
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1876
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1877
1877
|
"""
|
|
1878
1878
|
|
|
1879
1879
|
@check_mask_along_axis_iid
|
|
@@ -1940,7 +1940,7 @@ class MelScale(AudioTensorOperation):
|
|
|
1940
1940
|
|
|
1941
1941
|
Tutorial Examples:
|
|
1942
1942
|
- `Illustration of audio transforms
|
|
1943
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
1943
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
1944
1944
|
"""
|
|
1945
1945
|
|
|
1946
1946
|
@check_mel_scale
|
|
@@ -2041,7 +2041,7 @@ class MelSpectrogram(AudioTensorOperation):
|
|
|
2041
2041
|
|
|
2042
2042
|
Tutorial Examples:
|
|
2043
2043
|
- `Illustration of audio transforms
|
|
2044
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2044
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2045
2045
|
"""
|
|
2046
2046
|
|
|
2047
2047
|
@check_mel_spectrogram
|
|
@@ -2128,7 +2128,7 @@ class MFCC(AudioTensorOperation):
|
|
|
2128
2128
|
|
|
2129
2129
|
Tutorial Examples:
|
|
2130
2130
|
- `Illustration of audio transforms
|
|
2131
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2131
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2132
2132
|
"""
|
|
2133
2133
|
|
|
2134
2134
|
@check_mfcc
|
|
@@ -2196,7 +2196,7 @@ class MuLawDecoding(AudioTensorOperation):
|
|
|
2196
2196
|
|
|
2197
2197
|
Tutorial Examples:
|
|
2198
2198
|
- `Illustration of audio transforms
|
|
2199
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2199
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2200
2200
|
"""
|
|
2201
2201
|
|
|
2202
2202
|
@check_mu_law_coding
|
|
@@ -2234,7 +2234,7 @@ class MuLawEncoding(AudioTensorOperation):
|
|
|
2234
2234
|
|
|
2235
2235
|
Tutorial Examples:
|
|
2236
2236
|
- `Illustration of audio transforms
|
|
2237
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2237
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2238
2238
|
"""
|
|
2239
2239
|
|
|
2240
2240
|
@check_mu_law_coding
|
|
@@ -2280,7 +2280,7 @@ class Overdrive(AudioTensorOperation):
|
|
|
2280
2280
|
|
|
2281
2281
|
Tutorial Examples:
|
|
2282
2282
|
- `Illustration of audio transforms
|
|
2283
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2283
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2284
2284
|
"""
|
|
2285
2285
|
|
|
2286
2286
|
@check_overdrive
|
|
@@ -2342,7 +2342,7 @@ class Phaser(AudioTensorOperation):
|
|
|
2342
2342
|
|
|
2343
2343
|
Tutorial Examples:
|
|
2344
2344
|
- `Illustration of audio transforms
|
|
2345
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2345
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2346
2346
|
"""
|
|
2347
2347
|
|
|
2348
2348
|
@check_phaser
|
|
@@ -2392,7 +2392,7 @@ class PhaseVocoder(AudioTensorOperation):
|
|
|
2392
2392
|
|
|
2393
2393
|
Tutorial Examples:
|
|
2394
2394
|
- `Illustration of audio transforms
|
|
2395
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2395
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2396
2396
|
"""
|
|
2397
2397
|
|
|
2398
2398
|
@check_phase_vocoder
|
|
@@ -2451,7 +2451,7 @@ class PitchShift(AudioTensorOperation):
|
|
|
2451
2451
|
|
|
2452
2452
|
Tutorial Examples:
|
|
2453
2453
|
- `Illustration of audio transforms
|
|
2454
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2454
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2455
2455
|
"""
|
|
2456
2456
|
|
|
2457
2457
|
@check_pitch_shift
|
|
@@ -2523,7 +2523,7 @@ class Resample(AudioTensorOperation):
|
|
|
2523
2523
|
|
|
2524
2524
|
Tutorial Examples:
|
|
2525
2525
|
- `Illustration of audio transforms
|
|
2526
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2526
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2527
2527
|
"""
|
|
2528
2528
|
|
|
2529
2529
|
@check_resample
|
|
@@ -2572,7 +2572,7 @@ class RiaaBiquad(AudioTensorOperation):
|
|
|
2572
2572
|
|
|
2573
2573
|
Tutorial Examples:
|
|
2574
2574
|
- `Illustration of audio transforms
|
|
2575
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2575
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2576
2576
|
"""
|
|
2577
2577
|
|
|
2578
2578
|
@check_riaa_biquad
|
|
@@ -2619,7 +2619,7 @@ class SlidingWindowCmn(AudioTensorOperation):
|
|
|
2619
2619
|
|
|
2620
2620
|
Tutorial Examples:
|
|
2621
2621
|
- `Illustration of audio transforms
|
|
2622
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2622
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2623
2623
|
"""
|
|
2624
2624
|
|
|
2625
2625
|
@check_sliding_window_cmn
|
|
@@ -2685,7 +2685,7 @@ class SpectralCentroid(TensorOperation):
|
|
|
2685
2685
|
|
|
2686
2686
|
Tutorial Examples:
|
|
2687
2687
|
- `Illustration of audio transforms
|
|
2688
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2688
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2689
2689
|
"""
|
|
2690
2690
|
|
|
2691
2691
|
@check_spectral_centroid
|
|
@@ -2758,7 +2758,7 @@ class Spectrogram(TensorOperation):
|
|
|
2758
2758
|
|
|
2759
2759
|
Tutorial Examples:
|
|
2760
2760
|
- `Illustration of audio transforms
|
|
2761
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2761
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2762
2762
|
"""
|
|
2763
2763
|
|
|
2764
2764
|
@check_spectrogram
|
|
@@ -2825,7 +2825,7 @@ class TimeMasking(AudioTensorOperation):
|
|
|
2825
2825
|
|
|
2826
2826
|
Tutorial Examples:
|
|
2827
2827
|
- `Illustration of audio transforms
|
|
2828
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2828
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2829
2829
|
|
|
2830
2830
|
.. image:: time_masking_original.png
|
|
2831
2831
|
|
|
@@ -2883,7 +2883,7 @@ class TimeStretch(AudioTensorOperation):
|
|
|
2883
2883
|
|
|
2884
2884
|
Tutorial Examples:
|
|
2885
2885
|
- `Illustration of audio transforms
|
|
2886
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2886
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2887
2887
|
|
|
2888
2888
|
.. image:: time_stretch_rate1.5.png
|
|
2889
2889
|
|
|
@@ -2943,7 +2943,7 @@ class TrebleBiquad(AudioTensorOperation):
|
|
|
2943
2943
|
|
|
2944
2944
|
Tutorial Examples:
|
|
2945
2945
|
- `Illustration of audio transforms
|
|
2946
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
2946
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
2947
2947
|
"""
|
|
2948
2948
|
|
|
2949
2949
|
@check_treble_biquad
|
|
@@ -3049,7 +3049,7 @@ class Vad(AudioTensorOperation):
|
|
|
3049
3049
|
|
|
3050
3050
|
Tutorial Examples:
|
|
3051
3051
|
- `Illustration of audio transforms
|
|
3052
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
3052
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
3053
3053
|
"""
|
|
3054
3054
|
|
|
3055
3055
|
@check_vad
|
|
@@ -3124,7 +3124,7 @@ class Vol(AudioTensorOperation):
|
|
|
3124
3124
|
|
|
3125
3125
|
Tutorial Examples:
|
|
3126
3126
|
- `Illustration of audio transforms
|
|
3127
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
3127
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/audio_gallery.html>`_
|
|
3128
3128
|
"""
|
|
3129
3129
|
|
|
3130
3130
|
@check_vol
|
|
@@ -47,6 +47,12 @@ class DSCallback:
|
|
|
47
47
|
...
|
|
48
48
|
... def ds_epoch_end(self, ds_run_context):
|
|
49
49
|
... print("callback: epoch end, we are in epoch", ds_run_context.cur_epoch_num, flush=True)
|
|
50
|
+
...
|
|
51
|
+
... def ds_step_begin(self, ds_run_context):
|
|
52
|
+
... print("callback: step begin, step", ds_run_context.cur_step_num_in_epoch, flush=True)
|
|
53
|
+
...
|
|
54
|
+
... def ds_step_end(self, ds_run_context):
|
|
55
|
+
... print("callback: step end, step", ds_run_context.cur_step_num_in_epoch, flush=True)
|
|
50
56
|
>>>
|
|
51
57
|
>>> dataset = ds.GeneratorDataset([1, 2], "col1", shuffle=False, num_parallel_workers=1)
|
|
52
58
|
>>> dataset = dataset.map(operations=lambda x: x, callbacks=PrintInfo())
|
|
@@ -58,8 +64,16 @@ class DSCallback:
|
|
|
58
64
|
... pass
|
|
59
65
|
callback: start dataset pipeline
|
|
60
66
|
callback: epoch begin, we are in epoch 1
|
|
67
|
+
callback: step begin, step 1
|
|
68
|
+
callback: step begin, step 2
|
|
69
|
+
callback: step end, step 1
|
|
70
|
+
callback: step end, step 2
|
|
61
71
|
callback: epoch end, we are in epoch 1
|
|
62
72
|
callback: epoch begin, we are in epoch 2
|
|
73
|
+
callback: step begin, step 1
|
|
74
|
+
callback: step begin, step 2
|
|
75
|
+
callback: step end, step 1
|
|
76
|
+
callback: step end, step 2
|
|
63
77
|
callback: epoch end, we are in epoch 2
|
|
64
78
|
"""
|
|
65
79
|
|
|
@@ -146,7 +160,7 @@ class WaitedDSCallback(Callback, DSCallback):
|
|
|
146
160
|
r"""
|
|
147
161
|
Abstract base class used to build dataset callback classes that are synchronized with the training callback class
|
|
148
162
|
`mindspore.train.Callback \
|
|
149
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
163
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/train/
|
|
150
164
|
mindspore.train.Callback.html#mindspore.train.Callback>`_ .
|
|
151
165
|
|
|
152
166
|
It can be used to execute a custom callback method before a step or an epoch, such as
|
|
@@ -157,7 +171,7 @@ class WaitedDSCallback(Callback, DSCallback):
|
|
|
157
171
|
`device_number` , `list_callback` , `cur_epoch_num` , `cur_step_num` , `dataset_sink_mode` ,
|
|
158
172
|
`net_outputs` , etc., see
|
|
159
173
|
`mindspore.train.Callback \
|
|
160
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
174
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/train/
|
|
161
175
|
mindspore.train.Callback.html#mindspore.train.Callback>`_ .
|
|
162
176
|
|
|
163
177
|
Users can obtain the dataset pipeline context through `ds_run_context` , including
|