2020-11-20 18:10:40 +00:00
|
|
|
import os
|
2021-03-05 16:50:21 +00:00
|
|
|
|
|
|
|
from six.moves import urllib
|
2020-11-20 18:10:40 +00:00
|
|
|
|
|
|
|
from pytorch_lightning.utilities import _module_available
|
|
|
|
|
2021-03-05 16:50:21 +00:00
|
|
|
# TorchVision hotfix https://github.com/pytorch/vision/issues/1938
|
|
|
|
opener = urllib.request.build_opener()
|
2021-07-26 11:37:35 +00:00
|
|
|
opener.addheaders = [("User-agent", "Mozilla/5.0")]
|
2021-03-05 16:50:21 +00:00
|
|
|
urllib.request.install_opener(opener)
|
|
|
|
|
2020-12-23 23:11:42 +00:00
|
|
|
_EXAMPLES_ROOT = os.path.dirname(__file__)
|
|
|
|
_PACKAGE_ROOT = os.path.dirname(_EXAMPLES_ROOT)
|
2021-07-26 11:37:35 +00:00
|
|
|
_DATASETS_PATH = os.path.join(_PACKAGE_ROOT, "Datasets")
|
2020-11-20 18:10:40 +00:00
|
|
|
|
2020-12-23 23:11:42 +00:00
|
|
|
_DALI_AVAILABLE = _module_available("nvidia.dali")
|
2020-12-17 09:21:00 +00:00
|
|
|
|
|
|
|
LIGHTNING_LOGO = """
|
|
|
|
####
|
|
|
|
###########
|
|
|
|
####################
|
|
|
|
############################
|
|
|
|
#####################################
|
|
|
|
##############################################
|
|
|
|
######################### ###################
|
|
|
|
####################### ###################
|
|
|
|
#################### ####################
|
|
|
|
################## #####################
|
|
|
|
################ ######################
|
|
|
|
##################### #################
|
|
|
|
###################### ###################
|
|
|
|
##################### #####################
|
|
|
|
#################### #######################
|
|
|
|
################### #########################
|
|
|
|
##############################################
|
|
|
|
#####################################
|
|
|
|
############################
|
|
|
|
####################
|
|
|
|
##########
|
|
|
|
####
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
def nice_print(msg, last=False):
|
|
|
|
print()
|
|
|
|
print("\033[0;35m" + msg + "\033[0m")
|
|
|
|
if last:
|
|
|
|
print()
|
|
|
|
|
|
|
|
|
|
|
|
def cli_lightning_logo():
|
|
|
|
nice_print(LIGHTNING_LOGO)
|