@@ -199,7 +199,6 @@ def __init__(
199
199
:class:`~sagemaker.estimator.Framework` and
200
200
:class:`~sagemaker.estimator.EstimatorBase`.
201
201
"""
202
- logger .info ('Calling viskaria fork of python sdk.' )
203
202
validate_version_or_image_args (framework_version , py_version , image_uri )
204
203
if py_version == "py2" :
205
204
logger .warning (
@@ -209,9 +208,6 @@ def __init__(
209
208
self .py_version = py_version
210
209
211
210
if distribution is not None :
212
- logger .info (
213
- "viskaria distribution is: %s" , distribution
214
- )
215
211
instance_type = renamed_kwargs (
216
212
"train_instance_type" , "instance_type" , kwargs .get ("instance_type" ), kwargs
217
213
)
@@ -259,10 +255,7 @@ def _pytorch_distribution_configuration(self, distribution):
259
255
"""
260
256
distribution_config = {}
261
257
if "pytorchddp" in distribution :
262
- logger .info ("viskaria pytorchddp found in distribution" )
263
- pytorch_ddp_dict = distribution ["pytorchddp" ]
264
258
pytorch_ddp_enabled = distribution .get ("pytorchddp" ).get ("enabled" , False )
265
- logger .info ("viskaria pytorch_ddp_enabled is %s" , pytorch_ddp_enabled )
266
259
distribution_config [self .LAUNCH_PYTORCH_DDP_ENV_NAME ] = pytorch_ddp_enabled
267
260
else :
268
261
distribution_config = self ._distribution_configuration (distribution = distribution )
0 commit comments