Logging from Custom Docker Code in Sagemaker

Unlike in AWS Lambda code, in Sagemaker training jobs using custom Python code in Docker containers logging won’t work unless a stream handler is specifically added to the logger.

logger = logging.getLogger("__main__")
stream_handler = logging.StreamHandler(sys.stdout)
logger.addHandler(stream_handler)