super-gradients/src/super_gradients/common/environment/environment_config.py

import logging
import os
from os import environ

import pkg_resources

try:
    PKG_CHECKPOINTS_DIR = pkg_resources.resource_filename("checkpoints", "")
except Exception:
    os.makedirs(os.path.join(os.getcwd(), "checkpoints"), exist_ok=True)
    PKG_CHECKPOINTS_DIR = os.path.join(os.getcwd(), "checkpoints")

AWS_ENV_NAME = environ.get("ENVIRONMENT_NAME")

AWS_ENVIRONMENTS = ["development", "staging", "production"]
if AWS_ENV_NAME not in AWS_ENVIRONMENTS:
    if AWS_ENV_NAME is None:
        if AWS_ENV_NAME not in AWS_ENVIRONMENTS:
            print(
                f"You did not mention an AWS environment."
                f'You can set the environment variable ENVIRONMENT_NAME with one of the values: {",".join(AWS_ENVIRONMENTS)}'
            )
        else:
            print(
                f'Bad AWS environment name: {AWS_ENV_NAME}. Please set an environment variable named ENVIRONMENT_NAME '
                f'with one of the values: {",".join(AWS_ENVIRONMENTS)}'
            )

# Controlling the default logging level via environment variable
DEFAULT_LOGGING_LEVEL = environ.get("LOG_LEVEL", "INFO").upper()

logging.basicConfig(
    level=DEFAULT_LOGGING_LEVEL
)  # Set the default level for all libraries - including 3rd party packages

DDP_LOCAL_RANK = -1

INIT_TRAINER = False