fa6dffdaf7a65afc85577153a2e151096fbc1634,catalyst/utils/distributed.py,,is_apex_available,#,34

Before Change

    try:
        import apex  // noqa: F401
        from apex import amp  // noqa: F401
        return True
    except ImportError:
        return False

After Change

def is_apex_available() -> bool:
    env_apex = os.getenv("USE_APEX", "1") == "1"
    try:
        import apex  // noqa: F401
        from apex import amp  // noqa: F401
        return True and env_apex
    except ImportError:
        return False and env_apex

Instances