├── .coveragerc ├── .github ├── CODEOWNERS └── PULL_REQUEST_TEMPLATE.md ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── LICENSE-THIRD-PARTY ├── NOTICE ├── README.md ├── THIRD-PARTY ├── bin └── build_binaries.py ├── codecov.yml ├── compile_protobuf.py ├── config ├── buildspec.yml ├── buildspec_build_wheel.yml ├── buildspec_empty.yml ├── buildspec_pre_commit.yml ├── buildspec_py38plus.yml ├── buildspec_tensorflow_2_3.yml ├── buildspec_tensorflow_2_4.yml ├── buildspec_tensorflow_2_5.yml ├── buildspec_tensorflow_2_7.yml ├── buildspec_vanilla_framework_tests.yml ├── buildspec_xgboost_1_2_1.yml ├── buildspec_zero_code_change.yml ├── change_branch.sh ├── check_smdebug_install.sh ├── codecov.sh ├── delete_older_artifacts.sh ├── install_smdebug.sh ├── pipeline │ └── buildspec_unit_tests.yml ├── profiler │ ├── buildspec_profiler_sagemaker_pytorch_1_5_0_integration_tests.yml │ ├── buildspec_profiler_sagemaker_pytorch_1_6_0_integration_tests.yml │ ├── buildspec_profiler_sagemaker_pytorch_1_7_0_integration_tests.yml │ ├── buildspec_profiler_sagemaker_pytorch_1_8_0_integration_tests.yml │ ├── buildspec_profiler_sagemaker_tensorflow_2_2_1_integration_tests.yml │ ├── buildspec_profiler_sagemaker_tensorflow_2_3_1_integration_tests.yml │ ├── buildspec_profiler_sagemaker_tensorflow_2_4_1_integration_tests.yml │ ├── buildspec_profiler_sagemaker_tensorflow_2_5_0_integration_tests.yml │ ├── requirements.txt │ └── run_profiler_integration_tests.sh ├── protoc_downloader.sh ├── tests.sh └── upload_on_end.sh ├── docs ├── analysis.md ├── api.md ├── distributed_training.md ├── env_var.md ├── mxnet.md ├── pytorch.md ├── resources │ ├── results_resnet.png │ ├── tensorflow_rules_loss.png │ └── xgboost_feature_importance.png ├── sagemaker.md ├── tensorflow.md └── xgboost.md ├── examples ├── mxnet │ ├── README.md │ └── scripts │ │ └── mxnet_fashionmnist_w_g_b.py ├── pytorch │ ├── README.md │ ├── scripts │ │ ├── pytorch_hook_demos.py │ │ ├── simple.py │ │ └── torch_resnet.py │ └── zero_code_change_examples │ │ ├── horovod_mnist.py │ │ └── smdataparallel_mnist.py ├── tensorflow │ ├── README.md │ ├── local │ │ ├── horovod_keras_mnist.py │ │ ├── mnist.py │ │ ├── simple.py │ │ └── tf_keras_resnet.py │ ├── sagemaker_byoc │ │ ├── horovod_keras_mnist.py │ │ ├── mnist.py │ │ ├── simple.py │ │ └── tf_keras_resnet.py │ └── sagemaker_official_container │ │ ├── horovod_estimator_mnist.py │ │ ├── horovod_keras_mnist.py │ │ ├── mnist.py │ │ ├── simple.py │ │ └── tf_keras_resnet.py ├── tensorflow2 │ ├── README.md │ └── scripts │ │ ├── smdataparallel_mnist_tf2.py │ │ ├── tf2_save_metrics.py │ │ ├── tf_keras_fit_eager.py │ │ ├── tf_keras_fit_hvd.py │ │ ├── tf_keras_fit_non_eager.py │ │ ├── tf_keras_gradienttape.py │ │ └── tf_save_metrics_gradient_tape.py └── xgboost │ └── README.md ├── setup.cfg ├── setup.py ├── smdebug ├── __init__.py ├── _version.py ├── analysis │ ├── __init__.py │ └── utils.py ├── core │ ├── __init__.py │ ├── access_layer │ │ ├── __init__.py │ │ ├── base.py │ │ ├── file.py │ │ ├── s3.py │ │ ├── s3handler.py │ │ └── utils.py │ ├── collection.py │ ├── collection_manager.py │ ├── config_constants.py │ ├── config_validator.py │ ├── error_handling_agent.py │ ├── hook.py │ ├── hook_utils.py │ ├── index_reader.py │ ├── json_config.py │ ├── locations.py │ ├── logger.py │ ├── modes.py │ ├── reader.py │ ├── reduction_config.py │ ├── reductions.py │ ├── s3_utils.py │ ├── sagemaker_utils.py │ ├── save_config.py │ ├── singleton_utils.py │ ├── state_store.py │ ├── tensor.py │ ├── tfevent │ │ ├── __init__.py │ │ ├── event_file_reader.py │ │ ├── event_file_writer.py │ │ ├── events_writer.py │ │ ├── index_file_writer.py │ │ ├── proto │ │ │ ├── __init__.py │ │ │ ├── attr_value.proto │ │ │ ├── event.proto │ │ │ ├── graph.proto │ │ │ ├── node_def.proto │ │ │ ├── resource_handle.proto │ │ │ ├── summary.proto │ │ │ ├── tensor.proto │ │ │ ├── tensor_shape.proto │ │ │ ├── types.proto │ │ │ └── versions.proto │ │ ├── summary.py │ │ ├── summary_to_event.txt │ │ ├── timeline_file_writer.py │ │ └── util.py │ ├── tfrecord │ │ ├── __init__.py │ │ ├── _crc32c.py │ │ ├── record_reader.py │ │ ├── record_writer.py │ │ └── tensor_reader.py │ ├── utils.py │ └── writer.py ├── exceptions.py ├── mxnet │ ├── __init__.py │ ├── collection.py │ ├── graph.py │ ├── hook.py │ ├── singleton_utils.py │ └── utils.py ├── profiler │ ├── __init__.py │ ├── algorithm_metrics_reader.py │ ├── analysis │ │ ├── __init__.py │ │ ├── notebook_utils │ │ │ ├── __init__.py │ │ │ ├── heatmap.py │ │ │ ├── metrics_histogram.py │ │ │ ├── step_histogram.py │ │ │ ├── step_timeline_chart.py │ │ │ ├── timeline_charts.py │ │ │ └── training_job.py │ │ ├── python_profile_analysis.py │ │ ├── python_stats_reader.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── merge_timelines.py │ │ │ ├── pandas_data_analysis.py │ │ │ ├── profiler_data_to_pandas.py │ │ │ ├── python_profile_analysis_utils.py │ │ │ └── pytorch_dataloader_analysis.py │ ├── hvd_trace_file_rotation.py │ ├── metrics_reader_base.py │ ├── profiler_config.py │ ├── profiler_config_parser.py │ ├── profiler_constants.py │ ├── python_profile_utils.py │ ├── python_profiler.py │ ├── system_metrics_reader.py │ ├── system_profiler_file_parser.py │ ├── tf_profiler_parser.py │ ├── trace_event_file_parser.py │ └── utils.py ├── pytorch │ ├── __init__.py │ ├── collection.py │ ├── hook.py │ ├── singleton_utils.py │ └── utils.py ├── rules │ ├── __init__.py │ ├── action │ │ ├── __init__.py │ │ ├── action.py │ │ ├── message_action.py │ │ └── stop_training_action.py │ ├── req_tensors.py │ ├── rule.py │ └── rule_invoker.py ├── tensorflow │ ├── __init__.py │ ├── base_hook.py │ ├── callable_cache.py │ ├── collection.py │ ├── constants.py │ ├── keras.py │ ├── reductions.py │ ├── session.py │ ├── singleton_utils.py │ ├── tensor_ref.py │ ├── utils.py │ └── version_check.py ├── trials │ ├── __init__.py │ ├── local_trial.py │ ├── profiler_trial.py │ ├── s3_trial.py │ ├── trial.py │ └── utils.py └── xgboost │ ├── __init__.py │ ├── collection.py │ ├── constants.py │ ├── hook.py │ ├── singleton_utils.py │ └── utils.py └── tests ├── __init__.py ├── analysis ├── __init__.py ├── conftest.py ├── exceptions │ ├── __init__.py │ └── test_exceptions.py ├── rules │ ├── __init__.py │ └── test_rule_no_refresh.py ├── tensors │ ├── __init__.py │ └── test_refresh.py ├── trials │ ├── __init__.py │ ├── test_create.py │ ├── test_has_passed_step_scenarios.py │ ├── test_load_collections.py │ ├── test_local.py │ ├── test_modes.py │ ├── test_refresh.py │ ├── test_s3.py │ └── test_tensors_api.py └── utils.py ├── conftest.py ├── constants.py ├── core ├── __init__.py ├── json_configs │ ├── all_params.json │ ├── case_insensitive_profiler_config_parser.json │ ├── complete_profiler_config_parser.json │ ├── file_interval_rotation_profiler_config_parser.json │ ├── file_open_fail_profiler_config_parser.json │ ├── file_size_rotation_profiler_config_parser.json │ ├── hvd_rotation_profiler_config_parser.json │ ├── invalid_profiler_config_parser.json │ ├── invalid_string_data_profiler_config_parser.json │ ├── new_step_profiler_config_parser.json │ ├── new_time_profiler_config_parser.json │ ├── old_step_profiler_config_parser.json │ ├── old_time_profiler_config_parser.json │ ├── simple_profiler_config_parser.json │ ├── string_data_profiler_config_parser.json │ ├── test_native_tf2_cprofile_profiler_config_parser.json │ ├── test_native_tf2_pyinstrument_profiler_config_parser.json │ ├── test_pytorch_profiler_config_parser.json │ ├── test_tf2_profiler_config_parser_by_step.json │ ├── test_tf2_profiler_config_parser_by_time.json │ └── user_disabled_profile_config_parser.json ├── test_collections.py ├── test_error_handling_agent.py ├── test_handler.py ├── test_index.py ├── test_index_reader.py ├── test_index_utils.py ├── test_logger.py ├── test_make_numpy_array.py ├── test_modes.py ├── test_numpy.py ├── test_paths.py ├── test_reduction_config.py ├── test_save_config.py ├── test_state_store.py ├── test_timeline_writer.py ├── test_training_end.py ├── test_utils.py └── utils.py ├── fixtures ├── __init__.py ├── error_handling_agent.py └── tf2_training.py ├── mxnet ├── __init__.py ├── mnist_gluon_model.py ├── test_custom_tensor.py ├── test_hook.py ├── test_hook_all_zero.py ├── test_hook_collections.py ├── test_hook_custom_collection.py ├── test_hook_loss_collection.py ├── test_hook_reduce_config.py ├── test_hook_save_all.py ├── test_hook_save_config.py ├── test_hook_save_scalar_mxnet.py ├── test_json_configs │ ├── checkpointconfig.json │ ├── test_hook_all_zero_hook.json │ ├── test_hook_from_json_config.json │ ├── test_hook_from_json_config_full.json │ ├── test_hook_reduce_config_hook.json │ ├── test_hook_save_all_hook.json │ ├── test_modes_hook.json │ └── test_save_config_hookjson_config.json ├── test_modes.py ├── test_spot_training.py └── test_training_end.py ├── profiler ├── __init__.py ├── core │ ├── __init__.py │ ├── test_algorithm_metric_readers.py │ ├── test_horovodprofiler_events.py │ ├── test_pandas_frames.py │ ├── test_profiler_config_parser.py │ ├── test_python_profiler.py │ ├── test_smtfprofiler_events.py │ ├── test_system_metric_reader.py │ ├── test_system_profiler_file_parser.py │ ├── test_tfprofiler_events.py │ ├── test_utils.py │ └── utils.py ├── pytorch │ ├── __init__.py │ ├── scripts │ │ └── pytorch_rnn.py │ ├── test_pytorch_profiler.py │ └── test_pytorch_profiler_rnn.py ├── resources │ ├── 1589314018481947000_1234-testhost_model_timeline.json │ ├── 1591160699.algo-1.json │ ├── 1591160699.algo-1.json.corrupt │ ├── __init__.py │ ├── horovod_timeline_small.json │ ├── horovod_timeline_traces │ │ └── framework │ │ │ └── pevents │ │ │ └── 2020070206 │ │ │ ├── 1593673051472361_88359-8c859046be41.ant.amazon.com_horovod_timeline.json │ │ │ └── 1593673051473228_88359-8c859046be41.ant.amazon.com_horovod_timeline.json │ ├── merge_traces │ │ └── framework │ │ │ └── pevents │ │ │ └── 2020071317 │ │ │ ├── 1594662614815222_905-b88ab1bb8259_model_timeline.json │ │ │ ├── 1594662624558411_905-b88ab1bb8259_model_timeline.json │ │ │ └── 1594662624718694_905-b88ab1bb8259_pythontimeline.json │ ├── model_timeline_traces │ │ └── framework │ │ │ └── pevents │ │ │ └── 2020052602 │ │ │ ├── 1590461127873222_0001_model_timeline.json │ │ │ ├── 1590461139923994_0001_model_timeline.json │ │ │ └── 1590461139949971_0001_model_timeline.json │ ├── profiler_config_parser_utils.py │ ├── test_traces │ │ ├── framework │ │ │ └── pevents │ │ │ │ └── 2020051923 │ │ │ │ ├── 1589930988000000_4487_0000_pythontimeline.json │ │ │ │ ├── 1589930989000000_4487_0000_pythontimeline.json │ │ │ │ ├── 1589930990000000_4487_0000_pythontimeline.json │ │ │ │ └── 1589930991000000_4487_0000_pythontimeline.json │ │ └── system │ │ │ └── incremental │ │ │ ├── 1591160699.algo-1.json │ │ │ ├── 1591748100.algo-1.json │ │ │ └── 1591748160.algo-1.json │ ├── tfprofiler_local_missing_metadata │ │ └── framework │ │ │ └── tensorflow │ │ │ └── detailed_profiling │ │ │ └── 2020080513 │ │ │ └── 000000000 │ │ │ └── plugins │ │ │ └── profile │ │ │ └── 2020_08_05_13_37_44 │ │ │ └── ip-172-31-19-241.trace.json.gz │ └── tfprofiler_timeline_traces │ │ └── framework │ │ └── tensorflow │ │ ├── cProfile │ │ ├── 1596585363775342.0_1596585373504789.0_151-algo-1_-1 │ │ │ └── python_stats │ │ ├── 1596585374963238.0_1596585376008161.0_151-algo-1_2 │ │ │ └── python_stats │ │ └── 1596585376012363.8_1596585376017246.5_151-algo-1_3 │ │ │ └── python_stats │ │ └── detailed_profiling │ │ └── 2020080513 │ │ └── 000000000 │ │ ├── 80807-8c859046be41.ant.amazon.com_1596659864545103_1596659864854168.metadata │ │ ├── events.out.tfevents.1596659864.8c859046be41.ant.amazon.com.profile-empty │ │ └── plugins │ │ └── profile │ │ └── 2020_08_05_13_37_44 │ │ ├── 8c859046be41.ant.amazon.com.input_pipeline.pb │ │ ├── 8c859046be41.ant.amazon.com.kernel_stats.pb │ │ ├── 8c859046be41.ant.amazon.com.overview_page.pb │ │ ├── 8c859046be41.ant.amazon.com.tensorflow_stats.pb │ │ └── 8c859046be41.ant.amazon.com.trace.json.gz └── tensorflow2 │ ├── test_native_tf2_profiler.py │ ├── test_tf2_profiler.py │ └── utils.py ├── pytorch ├── __init__.py ├── test_collection.py ├── test_config_validator_pytorch.py ├── test_data_parallel.py ├── test_distributed_training.py ├── test_get_hook.py ├── test_hook_paths.py ├── test_hook_save_scalar_pytorch.py ├── test_json_configs │ ├── test_hook_multi_collections.json │ ├── test_hook_reduction_config.json │ ├── test_hook_save_shape.json │ ├── test_hook_saveall.json │ ├── test_hook_weightsbiasgradients.json │ └── test_modes.json ├── test_loss.py ├── test_modes.py ├── test_no_name_clash.py ├── test_reduce_config.py ├── test_save_custom_tensor.py ├── test_simple_write.py └── utils.py ├── resources └── mxnet │ └── mnist_gluon_basic_hook_demo.py ├── rules ├── __init__.py └── action │ ├── __init__.py │ └── test_message_action.py ├── tensorflow ├── __init__.py ├── hooks │ ├── __init__.py │ ├── test_collection_defaults.py │ ├── test_dist_horovod.py │ ├── test_estimator_modes.py │ ├── test_json_configs │ │ ├── test_collection_defaults.json │ │ ├── test_hook_from_json_config.json │ │ ├── test_mnist_local.json │ │ ├── test_multi_collection_match.json │ │ ├── test_only_weights_and_gradients.json │ │ ├── test_reductions.json │ │ ├── test_save_config.json │ │ ├── test_save_config_disable.json │ │ ├── test_save_config_modes_config_coll.json │ │ ├── test_save_config_modes_hook_config.json │ │ ├── test_save_config_skip_steps.json │ │ ├── test_save_config_start_and_end.json │ │ ├── test_simple_include.json │ │ ├── test_simple_include_regex.json │ │ └── test_write.json │ ├── test_losses.py │ ├── test_mirrored_strategy.py │ ├── test_reductions.py │ ├── test_save_all_full.py │ ├── test_save_config.py │ ├── test_session.py │ ├── test_simple_include.py │ ├── test_training_end.py │ ├── test_weights_gradients.py │ ├── test_write.py │ └── utils.py ├── keras │ ├── __init__.py │ ├── test_keras.py │ ├── test_keras_lstm.py │ └── test_keras_mirrored.py ├── test_hook_save_scalar_tensorflow.py ├── test_keras_to_estimator.py ├── test_sagemaker.py ├── test_tf_collections.py ├── test_utils.py └── utils.py ├── tensorflow2 ├── __init__.py ├── test_concat_layer.py ├── test_config_validator_tensorflow.py ├── test_embedding_grad.py ├── test_estimator.py ├── test_functional_model.py ├── test_get_hook.py ├── test_grad_tape_tf_function.py ├── test_keras.py ├── test_keras_lstm.py ├── test_keras_mirrored.py ├── test_mixed_precision_training.py ├── test_model_subclassing.py ├── test_should_save_tensor.py ├── test_support_dicts.py ├── test_tensorflow2_datatypes.py └── utils.py ├── utils.py ├── xgboost ├── __init__.py ├── json_config.py ├── run_xgboost_model.py ├── test_hook.py ├── test_hook_save_scalar_xgboost.py ├── test_utils.py └── test_xgboost_error_handling_agent.py └── zero_code_change ├── README.md ├── __init__.py ├── horovod_tests ├── __init__.py ├── constants.py ├── pytorch │ ├── __init__.py │ └── test_hvd.py ├── tensorflow │ ├── __init__.py │ ├── test_estimator.py │ └── test_keras.py ├── tensorflow2 │ └── test_keras_fit.py ├── tf_utils.py └── utils.py ├── pt_utils.py ├── smdataparallel_tests ├── __init__.py ├── constants.py ├── pytorch │ ├── __init__.py │ └── test_smdataparallel.py ├── tensorflow2 │ ├── __init__.py │ └── test_tf2_smdataparallel.py └── utils.py ├── test_mxnet_error_handling_agent.py ├── test_mxnet_gluon_integration.py ├── test_pytorch_disengage_smdebug.py ├── test_pytorch_error_handling_agent.py ├── test_pytorch_integration.py ├── test_pytorch_multiprocessing.py ├── test_tensorflow2_bert.py ├── test_tensorflow2_error_handling_agent.py ├── test_tensorflow2_gradtape_integration.py ├── test_tensorflow2_integration.py ├── test_tensorflow_disengage_smdebug.py ├── test_tensorflow_error_handling_agent.py ├── test_tensorflow_integration.py ├── test_training_with_no_grad_updates.py ├── tf_utils.py └── utils.py /.coveragerc: -------------------------------------------------------------------------------- 1 | [run] 2 | parallel=true 3 | -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE-THIRD-PARTY: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/LICENSE-THIRD-PARTY -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/README.md -------------------------------------------------------------------------------- /THIRD-PARTY: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/THIRD-PARTY -------------------------------------------------------------------------------- /bin/build_binaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/bin/build_binaries.py -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/codecov.yml -------------------------------------------------------------------------------- /compile_protobuf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/compile_protobuf.py -------------------------------------------------------------------------------- /config/buildspec.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec.yml -------------------------------------------------------------------------------- /config/buildspec_build_wheel.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_build_wheel.yml -------------------------------------------------------------------------------- /config/buildspec_empty.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_empty.yml -------------------------------------------------------------------------------- /config/buildspec_pre_commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_pre_commit.yml -------------------------------------------------------------------------------- /config/buildspec_py38plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_py38plus.yml -------------------------------------------------------------------------------- /config/buildspec_tensorflow_2_3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_tensorflow_2_3.yml -------------------------------------------------------------------------------- /config/buildspec_tensorflow_2_4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_tensorflow_2_4.yml -------------------------------------------------------------------------------- /config/buildspec_tensorflow_2_5.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_tensorflow_2_5.yml -------------------------------------------------------------------------------- /config/buildspec_tensorflow_2_7.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_tensorflow_2_7.yml -------------------------------------------------------------------------------- /config/buildspec_vanilla_framework_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_vanilla_framework_tests.yml -------------------------------------------------------------------------------- /config/buildspec_xgboost_1_2_1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_xgboost_1_2_1.yml -------------------------------------------------------------------------------- /config/buildspec_zero_code_change.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/buildspec_zero_code_change.yml -------------------------------------------------------------------------------- /config/change_branch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/change_branch.sh -------------------------------------------------------------------------------- /config/check_smdebug_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/check_smdebug_install.sh -------------------------------------------------------------------------------- /config/codecov.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/codecov.sh -------------------------------------------------------------------------------- /config/delete_older_artifacts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/delete_older_artifacts.sh -------------------------------------------------------------------------------- /config/install_smdebug.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/install_smdebug.sh -------------------------------------------------------------------------------- /config/pipeline/buildspec_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/pipeline/buildspec_unit_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_pytorch_1_5_0_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_pytorch_1_5_0_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_pytorch_1_6_0_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_pytorch_1_6_0_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_pytorch_1_7_0_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_pytorch_1_7_0_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_pytorch_1_8_0_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_pytorch_1_8_0_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_tensorflow_2_2_1_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_tensorflow_2_2_1_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_tensorflow_2_3_1_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_tensorflow_2_3_1_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_tensorflow_2_4_1_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_tensorflow_2_4_1_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/buildspec_profiler_sagemaker_tensorflow_2_5_0_integration_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/buildspec_profiler_sagemaker_tensorflow_2_5_0_integration_tests.yml -------------------------------------------------------------------------------- /config/profiler/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/requirements.txt -------------------------------------------------------------------------------- /config/profiler/run_profiler_integration_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/profiler/run_profiler_integration_tests.sh -------------------------------------------------------------------------------- /config/protoc_downloader.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/protoc_downloader.sh -------------------------------------------------------------------------------- /config/tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/tests.sh -------------------------------------------------------------------------------- /config/upload_on_end.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/config/upload_on_end.sh -------------------------------------------------------------------------------- /docs/analysis.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/analysis.md -------------------------------------------------------------------------------- /docs/api.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/api.md -------------------------------------------------------------------------------- /docs/distributed_training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/distributed_training.md -------------------------------------------------------------------------------- /docs/env_var.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/env_var.md -------------------------------------------------------------------------------- /docs/mxnet.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/mxnet.md -------------------------------------------------------------------------------- /docs/pytorch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/pytorch.md -------------------------------------------------------------------------------- /docs/resources/results_resnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/resources/results_resnet.png -------------------------------------------------------------------------------- /docs/resources/tensorflow_rules_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/resources/tensorflow_rules_loss.png -------------------------------------------------------------------------------- /docs/resources/xgboost_feature_importance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/resources/xgboost_feature_importance.png -------------------------------------------------------------------------------- /docs/sagemaker.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/sagemaker.md -------------------------------------------------------------------------------- /docs/tensorflow.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/tensorflow.md -------------------------------------------------------------------------------- /docs/xgboost.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/docs/xgboost.md -------------------------------------------------------------------------------- /examples/mxnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/mxnet/README.md -------------------------------------------------------------------------------- /examples/mxnet/scripts/mxnet_fashionmnist_w_g_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/mxnet/scripts/mxnet_fashionmnist_w_g_b.py -------------------------------------------------------------------------------- /examples/pytorch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/README.md -------------------------------------------------------------------------------- /examples/pytorch/scripts/pytorch_hook_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/scripts/pytorch_hook_demos.py -------------------------------------------------------------------------------- /examples/pytorch/scripts/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/scripts/simple.py -------------------------------------------------------------------------------- /examples/pytorch/scripts/torch_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/scripts/torch_resnet.py -------------------------------------------------------------------------------- /examples/pytorch/zero_code_change_examples/horovod_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/zero_code_change_examples/horovod_mnist.py -------------------------------------------------------------------------------- /examples/pytorch/zero_code_change_examples/smdataparallel_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/pytorch/zero_code_change_examples/smdataparallel_mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/README.md -------------------------------------------------------------------------------- /examples/tensorflow/local/horovod_keras_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/local/horovod_keras_mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/local/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/local/mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/local/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/local/simple.py -------------------------------------------------------------------------------- /examples/tensorflow/local/tf_keras_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/local/tf_keras_resnet.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_byoc/horovod_keras_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_byoc/horovod_keras_mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_byoc/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_byoc/mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_byoc/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_byoc/simple.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_byoc/tf_keras_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_byoc/tf_keras_resnet.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_official_container/horovod_estimator_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_official_container/horovod_estimator_mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_official_container/horovod_keras_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_official_container/horovod_keras_mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_official_container/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_official_container/mnist.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_official_container/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_official_container/simple.py -------------------------------------------------------------------------------- /examples/tensorflow/sagemaker_official_container/tf_keras_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow/sagemaker_official_container/tf_keras_resnet.py -------------------------------------------------------------------------------- /examples/tensorflow2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/README.md -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/smdataparallel_mnist_tf2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/smdataparallel_mnist_tf2.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf2_save_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf2_save_metrics.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf_keras_fit_eager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf_keras_fit_eager.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf_keras_fit_hvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf_keras_fit_hvd.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf_keras_fit_non_eager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf_keras_fit_non_eager.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf_keras_gradienttape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf_keras_gradienttape.py -------------------------------------------------------------------------------- /examples/tensorflow2/scripts/tf_save_metrics_gradient_tape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/tensorflow2/scripts/tf_save_metrics_gradient_tape.py -------------------------------------------------------------------------------- /examples/xgboost/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/examples/xgboost/README.md -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/setup.py -------------------------------------------------------------------------------- /smdebug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/__init__.py -------------------------------------------------------------------------------- /smdebug/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.0.34" 2 | -------------------------------------------------------------------------------- /smdebug/analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/analysis/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/analysis/utils.py -------------------------------------------------------------------------------- /smdebug/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/core/access_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/__init__.py -------------------------------------------------------------------------------- /smdebug/core/access_layer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/base.py -------------------------------------------------------------------------------- /smdebug/core/access_layer/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/file.py -------------------------------------------------------------------------------- /smdebug/core/access_layer/s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/s3.py -------------------------------------------------------------------------------- /smdebug/core/access_layer/s3handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/s3handler.py -------------------------------------------------------------------------------- /smdebug/core/access_layer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/access_layer/utils.py -------------------------------------------------------------------------------- /smdebug/core/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/collection.py -------------------------------------------------------------------------------- /smdebug/core/collection_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/collection_manager.py -------------------------------------------------------------------------------- /smdebug/core/config_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/config_constants.py -------------------------------------------------------------------------------- /smdebug/core/config_validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/config_validator.py -------------------------------------------------------------------------------- /smdebug/core/error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/error_handling_agent.py -------------------------------------------------------------------------------- /smdebug/core/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/hook.py -------------------------------------------------------------------------------- /smdebug/core/hook_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/hook_utils.py -------------------------------------------------------------------------------- /smdebug/core/index_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/index_reader.py -------------------------------------------------------------------------------- /smdebug/core/json_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/json_config.py -------------------------------------------------------------------------------- /smdebug/core/locations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/locations.py -------------------------------------------------------------------------------- /smdebug/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/logger.py -------------------------------------------------------------------------------- /smdebug/core/modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/modes.py -------------------------------------------------------------------------------- /smdebug/core/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/reader.py -------------------------------------------------------------------------------- /smdebug/core/reduction_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/reduction_config.py -------------------------------------------------------------------------------- /smdebug/core/reductions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/reductions.py -------------------------------------------------------------------------------- /smdebug/core/s3_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/s3_utils.py -------------------------------------------------------------------------------- /smdebug/core/sagemaker_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/sagemaker_utils.py -------------------------------------------------------------------------------- /smdebug/core/save_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/save_config.py -------------------------------------------------------------------------------- /smdebug/core/singleton_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/singleton_utils.py -------------------------------------------------------------------------------- /smdebug/core/state_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/state_store.py -------------------------------------------------------------------------------- /smdebug/core/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tensor.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/__init__.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/event_file_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/event_file_reader.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/event_file_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/event_file_writer.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/events_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/events_writer.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/index_file_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/index_file_writer.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/attr_value.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/attr_value.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/event.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/event.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/graph.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/graph.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/node_def.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/node_def.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/resource_handle.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/resource_handle.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/summary.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/summary.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/tensor.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/tensor.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/tensor_shape.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/tensor_shape.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/types.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/types.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/proto/versions.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/proto/versions.proto -------------------------------------------------------------------------------- /smdebug/core/tfevent/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/summary.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/summary_to_event.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/summary_to_event.txt -------------------------------------------------------------------------------- /smdebug/core/tfevent/timeline_file_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/timeline_file_writer.py -------------------------------------------------------------------------------- /smdebug/core/tfevent/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfevent/util.py -------------------------------------------------------------------------------- /smdebug/core/tfrecord/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/core/tfrecord/_crc32c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfrecord/_crc32c.py -------------------------------------------------------------------------------- /smdebug/core/tfrecord/record_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfrecord/record_reader.py -------------------------------------------------------------------------------- /smdebug/core/tfrecord/record_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfrecord/record_writer.py -------------------------------------------------------------------------------- /smdebug/core/tfrecord/tensor_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/tfrecord/tensor_reader.py -------------------------------------------------------------------------------- /smdebug/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/utils.py -------------------------------------------------------------------------------- /smdebug/core/writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/core/writer.py -------------------------------------------------------------------------------- /smdebug/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/exceptions.py -------------------------------------------------------------------------------- /smdebug/mxnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/__init__.py -------------------------------------------------------------------------------- /smdebug/mxnet/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/collection.py -------------------------------------------------------------------------------- /smdebug/mxnet/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/graph.py -------------------------------------------------------------------------------- /smdebug/mxnet/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/hook.py -------------------------------------------------------------------------------- /smdebug/mxnet/singleton_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/singleton_utils.py -------------------------------------------------------------------------------- /smdebug/mxnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/mxnet/utils.py -------------------------------------------------------------------------------- /smdebug/profiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/__init__.py -------------------------------------------------------------------------------- /smdebug/profiler/algorithm_metrics_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/algorithm_metrics_reader.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/__init__.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/heatmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/heatmap.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/metrics_histogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/metrics_histogram.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/step_histogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/step_histogram.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/step_timeline_chart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/step_timeline_chart.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/timeline_charts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/timeline_charts.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/notebook_utils/training_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/notebook_utils/training_job.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/python_profile_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/python_profile_analysis.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/python_stats_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/python_stats_reader.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/merge_timelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/utils/merge_timelines.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/pandas_data_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/utils/pandas_data_analysis.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/profiler_data_to_pandas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/utils/profiler_data_to_pandas.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/python_profile_analysis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/utils/python_profile_analysis_utils.py -------------------------------------------------------------------------------- /smdebug/profiler/analysis/utils/pytorch_dataloader_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/analysis/utils/pytorch_dataloader_analysis.py -------------------------------------------------------------------------------- /smdebug/profiler/hvd_trace_file_rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/hvd_trace_file_rotation.py -------------------------------------------------------------------------------- /smdebug/profiler/metrics_reader_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/metrics_reader_base.py -------------------------------------------------------------------------------- /smdebug/profiler/profiler_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/profiler_config.py -------------------------------------------------------------------------------- /smdebug/profiler/profiler_config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/profiler_config_parser.py -------------------------------------------------------------------------------- /smdebug/profiler/profiler_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/profiler_constants.py -------------------------------------------------------------------------------- /smdebug/profiler/python_profile_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/python_profile_utils.py -------------------------------------------------------------------------------- /smdebug/profiler/python_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/python_profiler.py -------------------------------------------------------------------------------- /smdebug/profiler/system_metrics_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/system_metrics_reader.py -------------------------------------------------------------------------------- /smdebug/profiler/system_profiler_file_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/system_profiler_file_parser.py -------------------------------------------------------------------------------- /smdebug/profiler/tf_profiler_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/tf_profiler_parser.py -------------------------------------------------------------------------------- /smdebug/profiler/trace_event_file_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/trace_event_file_parser.py -------------------------------------------------------------------------------- /smdebug/profiler/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/profiler/utils.py -------------------------------------------------------------------------------- /smdebug/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/pytorch/__init__.py -------------------------------------------------------------------------------- /smdebug/pytorch/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/pytorch/collection.py -------------------------------------------------------------------------------- /smdebug/pytorch/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/pytorch/hook.py -------------------------------------------------------------------------------- /smdebug/pytorch/singleton_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/pytorch/singleton_utils.py -------------------------------------------------------------------------------- /smdebug/pytorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/pytorch/utils.py -------------------------------------------------------------------------------- /smdebug/rules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/__init__.py -------------------------------------------------------------------------------- /smdebug/rules/action/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/action/__init__.py -------------------------------------------------------------------------------- /smdebug/rules/action/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/action/action.py -------------------------------------------------------------------------------- /smdebug/rules/action/message_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/action/message_action.py -------------------------------------------------------------------------------- /smdebug/rules/action/stop_training_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/action/stop_training_action.py -------------------------------------------------------------------------------- /smdebug/rules/req_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/req_tensors.py -------------------------------------------------------------------------------- /smdebug/rules/rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/rule.py -------------------------------------------------------------------------------- /smdebug/rules/rule_invoker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/rules/rule_invoker.py -------------------------------------------------------------------------------- /smdebug/tensorflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/__init__.py -------------------------------------------------------------------------------- /smdebug/tensorflow/base_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/base_hook.py -------------------------------------------------------------------------------- /smdebug/tensorflow/callable_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/callable_cache.py -------------------------------------------------------------------------------- /smdebug/tensorflow/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/collection.py -------------------------------------------------------------------------------- /smdebug/tensorflow/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/constants.py -------------------------------------------------------------------------------- /smdebug/tensorflow/keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/keras.py -------------------------------------------------------------------------------- /smdebug/tensorflow/reductions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/reductions.py -------------------------------------------------------------------------------- /smdebug/tensorflow/session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/session.py -------------------------------------------------------------------------------- /smdebug/tensorflow/singleton_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/singleton_utils.py -------------------------------------------------------------------------------- /smdebug/tensorflow/tensor_ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/tensor_ref.py -------------------------------------------------------------------------------- /smdebug/tensorflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/utils.py -------------------------------------------------------------------------------- /smdebug/tensorflow/version_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/tensorflow/version_check.py -------------------------------------------------------------------------------- /smdebug/trials/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/__init__.py -------------------------------------------------------------------------------- /smdebug/trials/local_trial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/local_trial.py -------------------------------------------------------------------------------- /smdebug/trials/profiler_trial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/profiler_trial.py -------------------------------------------------------------------------------- /smdebug/trials/s3_trial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/s3_trial.py -------------------------------------------------------------------------------- /smdebug/trials/trial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/trial.py -------------------------------------------------------------------------------- /smdebug/trials/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/trials/utils.py -------------------------------------------------------------------------------- /smdebug/xgboost/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/xgboost/__init__.py -------------------------------------------------------------------------------- /smdebug/xgboost/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/xgboost/collection.py -------------------------------------------------------------------------------- /smdebug/xgboost/constants.py: -------------------------------------------------------------------------------- 1 | INCREASING_METRICS_REGEX = "^[a-zA-z]+-(auc|map)$" 2 | -------------------------------------------------------------------------------- /smdebug/xgboost/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/xgboost/hook.py -------------------------------------------------------------------------------- /smdebug/xgboost/singleton_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/xgboost/singleton_utils.py -------------------------------------------------------------------------------- /smdebug/xgboost/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/smdebug/xgboost/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/conftest.py -------------------------------------------------------------------------------- /tests/analysis/exceptions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/exceptions/test_exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/exceptions/test_exceptions.py -------------------------------------------------------------------------------- /tests/analysis/rules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/rules/test_rule_no_refresh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/rules/test_rule_no_refresh.py -------------------------------------------------------------------------------- /tests/analysis/tensors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/tensors/test_refresh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/tensors/test_refresh.py -------------------------------------------------------------------------------- /tests/analysis/trials/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/analysis/trials/test_create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_create.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_has_passed_step_scenarios.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_has_passed_step_scenarios.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_load_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_load_collections.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_local.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_modes.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_refresh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_refresh.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_s3.py -------------------------------------------------------------------------------- /tests/analysis/trials/test_tensors_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/trials/test_tensors_api.py -------------------------------------------------------------------------------- /tests/analysis/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/analysis/utils.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/constants.py: -------------------------------------------------------------------------------- 1 | TEST_DATASET_S3_PATH = "s3://smdebug-testing/datasets/" 2 | -------------------------------------------------------------------------------- /tests/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/core/json_configs/all_params.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/all_params.json -------------------------------------------------------------------------------- /tests/core/json_configs/case_insensitive_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/case_insensitive_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/complete_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/complete_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/file_interval_rotation_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/file_interval_rotation_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/file_open_fail_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/file_open_fail_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/file_size_rotation_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/file_size_rotation_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/hvd_rotation_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/hvd_rotation_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/invalid_profiler_config_parser.json: -------------------------------------------------------------------------------- 1 | {1, 2, 3, 4, 5} 2 | -------------------------------------------------------------------------------- /tests/core/json_configs/invalid_string_data_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/invalid_string_data_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/new_step_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/new_step_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/new_time_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/new_time_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/old_step_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/old_step_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/old_time_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/old_time_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/simple_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/simple_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/string_data_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/string_data_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/test_native_tf2_cprofile_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/test_native_tf2_cprofile_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/test_native_tf2_pyinstrument_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/test_native_tf2_pyinstrument_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/test_pytorch_profiler_config_parser.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/test_pytorch_profiler_config_parser.json -------------------------------------------------------------------------------- /tests/core/json_configs/test_tf2_profiler_config_parser_by_step.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/test_tf2_profiler_config_parser_by_step.json -------------------------------------------------------------------------------- /tests/core/json_configs/test_tf2_profiler_config_parser_by_time.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/json_configs/test_tf2_profiler_config_parser_by_time.json -------------------------------------------------------------------------------- /tests/core/json_configs/user_disabled_profile_config_parser.json: -------------------------------------------------------------------------------- 1 | { 2 | "ProfilingParameters": {} 3 | } 4 | -------------------------------------------------------------------------------- /tests/core/test_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_collections.py -------------------------------------------------------------------------------- /tests/core/test_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_error_handling_agent.py -------------------------------------------------------------------------------- /tests/core/test_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_handler.py -------------------------------------------------------------------------------- /tests/core/test_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_index.py -------------------------------------------------------------------------------- /tests/core/test_index_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_index_reader.py -------------------------------------------------------------------------------- /tests/core/test_index_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_index_utils.py -------------------------------------------------------------------------------- /tests/core/test_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_logger.py -------------------------------------------------------------------------------- /tests/core/test_make_numpy_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_make_numpy_array.py -------------------------------------------------------------------------------- /tests/core/test_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_modes.py -------------------------------------------------------------------------------- /tests/core/test_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_numpy.py -------------------------------------------------------------------------------- /tests/core/test_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_paths.py -------------------------------------------------------------------------------- /tests/core/test_reduction_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_reduction_config.py -------------------------------------------------------------------------------- /tests/core/test_save_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_save_config.py -------------------------------------------------------------------------------- /tests/core/test_state_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_state_store.py -------------------------------------------------------------------------------- /tests/core/test_timeline_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_timeline_writer.py -------------------------------------------------------------------------------- /tests/core/test_training_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_training_end.py -------------------------------------------------------------------------------- /tests/core/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/test_utils.py -------------------------------------------------------------------------------- /tests/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/core/utils.py -------------------------------------------------------------------------------- /tests/fixtures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fixtures/error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/fixtures/error_handling_agent.py -------------------------------------------------------------------------------- /tests/fixtures/tf2_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/fixtures/tf2_training.py -------------------------------------------------------------------------------- /tests/mxnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/mxnet/mnist_gluon_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/mnist_gluon_model.py -------------------------------------------------------------------------------- /tests/mxnet/test_custom_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_custom_tensor.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_all_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_all_zero.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_collections.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_custom_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_custom_collection.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_loss_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_loss_collection.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_reduce_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_reduce_config.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_save_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_save_all.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_save_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_save_config.py -------------------------------------------------------------------------------- /tests/mxnet/test_hook_save_scalar_mxnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_hook_save_scalar_mxnet.py -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/checkpointconfig.json: -------------------------------------------------------------------------------- 1 | { 2 | "LocalPath" : "/tmp/savedParams" 3 | } 4 | -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_hook_all_zero_hook.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_hook_all_zero_hook.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_hook_from_json_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_hook_from_json_config.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_hook_from_json_config_full.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_hook_from_json_config_full.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_hook_reduce_config_hook.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_hook_reduce_config_hook.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_hook_save_all_hook.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_hook_save_all_hook.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_modes_hook.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_modes_hook.json -------------------------------------------------------------------------------- /tests/mxnet/test_json_configs/test_save_config_hookjson_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_json_configs/test_save_config_hookjson_config.json -------------------------------------------------------------------------------- /tests/mxnet/test_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_modes.py -------------------------------------------------------------------------------- /tests/mxnet/test_spot_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_spot_training.py -------------------------------------------------------------------------------- /tests/mxnet/test_training_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/mxnet/test_training_end.py -------------------------------------------------------------------------------- /tests/profiler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/core/test_algorithm_metric_readers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_algorithm_metric_readers.py -------------------------------------------------------------------------------- /tests/profiler/core/test_horovodprofiler_events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_horovodprofiler_events.py -------------------------------------------------------------------------------- /tests/profiler/core/test_pandas_frames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_pandas_frames.py -------------------------------------------------------------------------------- /tests/profiler/core/test_profiler_config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_profiler_config_parser.py -------------------------------------------------------------------------------- /tests/profiler/core/test_python_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_python_profiler.py -------------------------------------------------------------------------------- /tests/profiler/core/test_smtfprofiler_events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_smtfprofiler_events.py -------------------------------------------------------------------------------- /tests/profiler/core/test_system_metric_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_system_metric_reader.py -------------------------------------------------------------------------------- /tests/profiler/core/test_system_profiler_file_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_system_profiler_file_parser.py -------------------------------------------------------------------------------- /tests/profiler/core/test_tfprofiler_events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_tfprofiler_events.py -------------------------------------------------------------------------------- /tests/profiler/core/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/test_utils.py -------------------------------------------------------------------------------- /tests/profiler/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/core/utils.py -------------------------------------------------------------------------------- /tests/profiler/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/pytorch/scripts/pytorch_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/pytorch/scripts/pytorch_rnn.py -------------------------------------------------------------------------------- /tests/profiler/pytorch/test_pytorch_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/pytorch/test_pytorch_profiler.py -------------------------------------------------------------------------------- /tests/profiler/pytorch/test_pytorch_profiler_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/pytorch/test_pytorch_profiler_rnn.py -------------------------------------------------------------------------------- /tests/profiler/resources/1589314018481947000_1234-testhost_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/1589314018481947000_1234-testhost_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/1591160699.algo-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/1591160699.algo-1.json -------------------------------------------------------------------------------- /tests/profiler/resources/1591160699.algo-1.json.corrupt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/1591160699.algo-1.json.corrupt -------------------------------------------------------------------------------- /tests/profiler/resources/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/resources/horovod_timeline_small.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/horovod_timeline_small.json -------------------------------------------------------------------------------- /tests/profiler/resources/horovod_timeline_traces/framework/pevents/2020070206/1593673051472361_88359-8c859046be41.ant.amazon.com_horovod_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/horovod_timeline_traces/framework/pevents/2020070206/1593673051472361_88359-8c859046be41.ant.amazon.com_horovod_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/horovod_timeline_traces/framework/pevents/2020070206/1593673051473228_88359-8c859046be41.ant.amazon.com_horovod_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/horovod_timeline_traces/framework/pevents/2020070206/1593673051473228_88359-8c859046be41.ant.amazon.com_horovod_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662614815222_905-b88ab1bb8259_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662614815222_905-b88ab1bb8259_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662624558411_905-b88ab1bb8259_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662624558411_905-b88ab1bb8259_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662624718694_905-b88ab1bb8259_pythontimeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/merge_traces/framework/pevents/2020071317/1594662624718694_905-b88ab1bb8259_pythontimeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461127873222_0001_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461127873222_0001_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461139923994_0001_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461139923994_0001_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461139949971_0001_model_timeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/model_timeline_traces/framework/pevents/2020052602/1590461139949971_0001_model_timeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/profiler_config_parser_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/profiler_config_parser_utils.py -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930988000000_4487_0000_pythontimeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930988000000_4487_0000_pythontimeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930989000000_4487_0000_pythontimeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930989000000_4487_0000_pythontimeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930990000000_4487_0000_pythontimeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930990000000_4487_0000_pythontimeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930991000000_4487_0000_pythontimeline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/framework/pevents/2020051923/1589930991000000_4487_0000_pythontimeline.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/system/incremental/1591160699.algo-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/system/incremental/1591160699.algo-1.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/system/incremental/1591748100.algo-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/system/incremental/1591748100.algo-1.json -------------------------------------------------------------------------------- /tests/profiler/resources/test_traces/system/incremental/1591748160.algo-1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/test_traces/system/incremental/1591748160.algo-1.json -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_local_missing_metadata/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/ip-172-31-19-241.trace.json.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_local_missing_metadata/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/ip-172-31-19-241.trace.json.gz -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585363775342.0_1596585373504789.0_151-algo-1_-1/python_stats: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585363775342.0_1596585373504789.0_151-algo-1_-1/python_stats -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585374963238.0_1596585376008161.0_151-algo-1_2/python_stats: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585374963238.0_1596585376008161.0_151-algo-1_2/python_stats -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585376012363.8_1596585376017246.5_151-algo-1_3/python_stats: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/cProfile/1596585376012363.8_1596585376017246.5_151-algo-1_3/python_stats -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/80807-8c859046be41.ant.amazon.com_1596659864545103_1596659864854168.metadata: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/events.out.tfevents.1596659864.8c859046be41.ant.amazon.com.profile-empty: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/events.out.tfevents.1596659864.8c859046be41.ant.amazon.com.profile-empty -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.input_pipeline.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.input_pipeline.pb -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.kernel_stats.pb: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.overview_page.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.overview_page.pb -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.tensorflow_stats.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.tensorflow_stats.pb -------------------------------------------------------------------------------- /tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.trace.json.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.trace.json.gz -------------------------------------------------------------------------------- /tests/profiler/tensorflow2/test_native_tf2_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/tensorflow2/test_native_tf2_profiler.py -------------------------------------------------------------------------------- /tests/profiler/tensorflow2/test_tf2_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/tensorflow2/test_tf2_profiler.py -------------------------------------------------------------------------------- /tests/profiler/tensorflow2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/profiler/tensorflow2/utils.py -------------------------------------------------------------------------------- /tests/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/pytorch/test_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_collection.py -------------------------------------------------------------------------------- /tests/pytorch/test_config_validator_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_config_validator_pytorch.py -------------------------------------------------------------------------------- /tests/pytorch/test_data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_data_parallel.py -------------------------------------------------------------------------------- /tests/pytorch/test_distributed_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_distributed_training.py -------------------------------------------------------------------------------- /tests/pytorch/test_get_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_get_hook.py -------------------------------------------------------------------------------- /tests/pytorch/test_hook_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_hook_paths.py -------------------------------------------------------------------------------- /tests/pytorch/test_hook_save_scalar_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_hook_save_scalar_pytorch.py -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_hook_multi_collections.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_hook_multi_collections.json -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_hook_reduction_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_hook_reduction_config.json -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_hook_save_shape.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_hook_save_shape.json -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_hook_saveall.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_hook_saveall.json -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_hook_weightsbiasgradients.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_hook_weightsbiasgradients.json -------------------------------------------------------------------------------- /tests/pytorch/test_json_configs/test_modes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_json_configs/test_modes.json -------------------------------------------------------------------------------- /tests/pytorch/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_loss.py -------------------------------------------------------------------------------- /tests/pytorch/test_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_modes.py -------------------------------------------------------------------------------- /tests/pytorch/test_no_name_clash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_no_name_clash.py -------------------------------------------------------------------------------- /tests/pytorch/test_reduce_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_reduce_config.py -------------------------------------------------------------------------------- /tests/pytorch/test_save_custom_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_save_custom_tensor.py -------------------------------------------------------------------------------- /tests/pytorch/test_simple_write.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/test_simple_write.py -------------------------------------------------------------------------------- /tests/pytorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/pytorch/utils.py -------------------------------------------------------------------------------- /tests/resources/mxnet/mnist_gluon_basic_hook_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/resources/mxnet/mnist_gluon_basic_hook_demo.py -------------------------------------------------------------------------------- /tests/rules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/rules/action/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/rules/action/test_message_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/rules/action/test_message_action.py -------------------------------------------------------------------------------- /tests/tensorflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tensorflow/hooks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_collection_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_collection_defaults.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_dist_horovod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_dist_horovod.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_estimator_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_estimator_modes.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_collection_defaults.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_collection_defaults.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_hook_from_json_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_hook_from_json_config.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_mnist_local.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_mnist_local.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_multi_collection_match.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_multi_collection_match.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_only_weights_and_gradients.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_only_weights_and_gradients.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_reductions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_reductions.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config_disable.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config_disable.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config_modes_config_coll.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config_modes_config_coll.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config_modes_hook_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config_modes_hook_config.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config_skip_steps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config_skip_steps.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_save_config_start_and_end.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_save_config_start_and_end.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_simple_include.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_simple_include.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_simple_include_regex.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_simple_include_regex.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_json_configs/test_write.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_json_configs/test_write.json -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_losses.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_mirrored_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_mirrored_strategy.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_reductions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_reductions.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_save_all_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_save_all_full.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_save_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_save_config.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_session.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_simple_include.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_simple_include.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_training_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_training_end.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_weights_gradients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_weights_gradients.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/test_write.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/test_write.py -------------------------------------------------------------------------------- /tests/tensorflow/hooks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/hooks/utils.py -------------------------------------------------------------------------------- /tests/tensorflow/keras/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tensorflow/keras/test_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/keras/test_keras.py -------------------------------------------------------------------------------- /tests/tensorflow/keras/test_keras_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/keras/test_keras_lstm.py -------------------------------------------------------------------------------- /tests/tensorflow/keras/test_keras_mirrored.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/keras/test_keras_mirrored.py -------------------------------------------------------------------------------- /tests/tensorflow/test_hook_save_scalar_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/test_hook_save_scalar_tensorflow.py -------------------------------------------------------------------------------- /tests/tensorflow/test_keras_to_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/test_keras_to_estimator.py -------------------------------------------------------------------------------- /tests/tensorflow/test_sagemaker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/test_sagemaker.py -------------------------------------------------------------------------------- /tests/tensorflow/test_tf_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/test_tf_collections.py -------------------------------------------------------------------------------- /tests/tensorflow/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/test_utils.py -------------------------------------------------------------------------------- /tests/tensorflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow/utils.py -------------------------------------------------------------------------------- /tests/tensorflow2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tensorflow2/test_concat_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_concat_layer.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_config_validator_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_config_validator_tensorflow.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_embedding_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_embedding_grad.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_estimator.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_functional_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_functional_model.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_get_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_get_hook.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_grad_tape_tf_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_grad_tape_tf_function.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_keras.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_keras_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_keras_lstm.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_keras_mirrored.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_keras_mirrored.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_mixed_precision_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_mixed_precision_training.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_model_subclassing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_model_subclassing.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_should_save_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_should_save_tensor.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_support_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_support_dicts.py -------------------------------------------------------------------------------- /tests/tensorflow2/test_tensorflow2_datatypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/test_tensorflow2_datatypes.py -------------------------------------------------------------------------------- /tests/tensorflow2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/tensorflow2/utils.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/utils.py -------------------------------------------------------------------------------- /tests/xgboost/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/xgboost/json_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/json_config.py -------------------------------------------------------------------------------- /tests/xgboost/run_xgboost_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/run_xgboost_model.py -------------------------------------------------------------------------------- /tests/xgboost/test_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/test_hook.py -------------------------------------------------------------------------------- /tests/xgboost/test_hook_save_scalar_xgboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/test_hook_save_scalar_xgboost.py -------------------------------------------------------------------------------- /tests/xgboost/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/test_utils.py -------------------------------------------------------------------------------- /tests/xgboost/test_xgboost_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/xgboost/test_xgboost_error_handling_agent.py -------------------------------------------------------------------------------- /tests/zero_code_change/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/README.md -------------------------------------------------------------------------------- /tests/zero_code_change/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/constants.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/pytorch/test_hvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/pytorch/test_hvd.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/tensorflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/tensorflow/test_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/tensorflow/test_estimator.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/tensorflow/test_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/tensorflow/test_keras.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/tensorflow2/test_keras_fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/tensorflow2/test_keras_fit.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/tf_utils.py -------------------------------------------------------------------------------- /tests/zero_code_change/horovod_tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/horovod_tests/utils.py -------------------------------------------------------------------------------- /tests/zero_code_change/pt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/pt_utils.py -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/smdataparallel_tests/constants.py -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/pytorch/test_smdataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/smdataparallel_tests/pytorch/test_smdataparallel.py -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/tensorflow2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/tensorflow2/test_tf2_smdataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/smdataparallel_tests/tensorflow2/test_tf2_smdataparallel.py -------------------------------------------------------------------------------- /tests/zero_code_change/smdataparallel_tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/smdataparallel_tests/utils.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_mxnet_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_mxnet_error_handling_agent.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_mxnet_gluon_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_mxnet_gluon_integration.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_pytorch_disengage_smdebug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_pytorch_disengage_smdebug.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_pytorch_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_pytorch_error_handling_agent.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_pytorch_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_pytorch_integration.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_pytorch_multiprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_pytorch_multiprocessing.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow2_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow2_bert.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow2_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow2_error_handling_agent.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow2_gradtape_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow2_gradtape_integration.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow2_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow2_integration.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow_disengage_smdebug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow_disengage_smdebug.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow_error_handling_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow_error_handling_agent.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_tensorflow_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_tensorflow_integration.py -------------------------------------------------------------------------------- /tests/zero_code_change/test_training_with_no_grad_updates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/test_training_with_no_grad_updates.py -------------------------------------------------------------------------------- /tests/zero_code_change/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/tf_utils.py -------------------------------------------------------------------------------- /tests/zero_code_change/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/sagemaker-debugger/HEAD/tests/zero_code_change/utils.py --------------------------------------------------------------------------------