Skip to content

Commit

Permalink
Update readme, setup.py and updating the version. (#17)
Browse files Browse the repository at this point in the history
* Updating the README file.

* Updated the setup.py with long description and updated the version

* UPdating the README to use Python SDK instead of PySDK

* Added the license string to setup.py

* Addressed the review comments.
  • Loading branch information
leleamol authored and ddavydenko committed Dec 3, 2019
1 parent 3b54c79 commit 6d567a8
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 19 deletions.
19 changes: 8 additions & 11 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
## SageMaker-Debugger-RulesConfig
## Amazon SageMaker Debugger RulesConfig

Sagemaker-debugger is an upcoming AWS service designed to be a debugger for machine learning models. It lets you go beyond just looking at scalars like losses and accuracies during training and gives you full visibility into all tensors 'flowing through the graph' during training or inference.
Amazon SageMaker Debugger is designed to be a debugger for machine learning models. It lets you go beyond just looking at scalars like losses and accuracies during training and gives you full visibility into all tensors 'flowing through the graph' during training or inference.

SageMaker Debugger RulesConfig provides a mapping of builtin rules with default configurations. These configurations will affect both DebugHookConfig and DebugRuleConfigurations in the SageMaker Python SDK.
Amazon SageMaker Debugger RulesConfig provides a mapping of builtin rules with default configurations. These configurations will affect both DebugHookConfig and DebugRuleConfigurations in the Amazon SageMaker Python SDK.

This library, intended to be used with SageMaker PySDK, helps you specify builtin rules without worrying about any details or tweak the configuration of builtin rules. These builtin rules are available in SageMaker.
This library, intended to be used with Amazon SageMaker Python SDK, helps you specify builtin rules without worrying about any details or tweak the configuration of builtin rules. These builtin rules are available in Amazon SageMaker.

SageMaker Debugger Rulesconfig package can be used with SageMaker Debugger or as stand-alone rule config retriever too. In addition to retrieving builtin rules, configuration for common collections can be retrieved as well.
Amazon SageMaker Debugger Rulesconfig package can be used with Amazon SageMaker Debugger or as stand-alone rule config retriever. In addition to retrieving builtin rules, configuration for common collections can be retrieved as well.

Example: Vanilla builtin rule without customization

Expand All @@ -20,7 +20,7 @@ my_estimator = Estimator(
)
```

Example: Builtin rule with customization
Example: Builtin rule with customization. For more details please refer to [Amazon SageMaker Python SDK](https://github.com/aws/sagemaker-python-sdk) documentation.

```
my_estimator = Estimator(
Expand All @@ -29,18 +29,16 @@ my_estimator = Estimator(
Rule.sagemaker(vanishing_gradient()),
Rule.sagemaker(
base_config=weight_update_ratio(),
instance_type="t2.micro",
name="my_wup_rule_name",
container_local_path="/local/path",
s3_output_path="s3://uri",
volume_size_in_gb=100,
rule_parameters={
"param1": "value1",
"param2": "value2"
},
collection_configurations=[
collections_to_save=[
CollectionConfiguration(
name="my_name", # Required
name="my_name", # Required. If specified, debugger will collect tensors for this collection. Users may have to update rule_parameters above to run the rule on right tensors.
parameters= {
"param1": "value1",
"param2": "value2"
Expand Down Expand Up @@ -75,4 +73,3 @@ my_estimator = Estimator(
## License

This project is licensed under the Apache-2.0 License.

19 changes: 12 additions & 7 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,32 +1,37 @@
# Standard Library
import os
import sys
import setuptools

with open("README.md", "r") as fh:
long_description = fh.read()
long_description = (
"This library, intended to be used with [Amazon SageMaker Python SDK]("
"https://github.com/aws/sagemaker-python-sdk), helps you specify builtin "
"rules without worrying about any details or tweak the configuration of builtin rules. These "
"builtin rules are available in Amazon SageMaker."
"Amazon SageMaker Debugger Rulesconfig package can be used with Amazon SageMaker Debugger or as"
"stand-alone rule config retriever. In addition to retrieving builtin rules, configuration for common collections can be retrieved as well."
)

exec(open("smdebug_rulesconfig/_version.py").read())
CURRENT_VERSION = __version__


def build_rule_config_package(version):
setuptools.setup(
name="smdebug_rulesconfig",
version=version,
author="AWS DeepLearning Team",
description="SMDebug RulesConfig",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/awslabs/sagemaker-debugger-rulesconfig",
packages=["smdebug_rulesconfig"],
package_data={'smdebug_rulesconfig' : ['rule_config_jsons/*.json']},
package_data={"smdebug_rulesconfig": ["rule_config_jsons/*.json"]},
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
],
python_requires=">=2.7",
license="Apache License Version 2.0",
)

build_rule_config_package(version=CURRENT_VERSION)

build_rule_config_package(version=CURRENT_VERSION)
2 changes: 1 addition & 1 deletion smdebug_rulesconfig/_version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.1.2"
__version__ = "0.1.3"

0 comments on commit 6d567a8

Please sign in to comment.