Python get_build_infoの例

コード例 #1

ファイル: parrots_wrapper.py プロジェクト: hhaAndroid/mmcv

def get_build_config():
    if TORCH_VERSION == 'parrots':
        from parrots.config import get_build_info
        return get_build_info()
    else:
        return torch.__config__.show()

コード例 #2

ファイル: parrots_wrapper.py プロジェクト: researchmm/soho

def get_build_config():
    if torch.__version__ == 'parrots':
        from parrots.config import get_build_info
        return get_build_info()
    else:
        return torch.__config__.show()