11 lines
321 B
Python
11 lines
321 B
Python
import os
|
|
|
|
from pytorch_lightning.utilities import _module_available
|
|
|
|
EXAMPLES_ROOT = os.path.dirname(__file__)
|
|
PACKAGE_ROOT = os.path.dirname(EXAMPLES_ROOT)
|
|
DATASETS_PATH = os.path.join(PACKAGE_ROOT, 'Datasets')
|
|
|
|
TORCHVISION_AVAILABLE = _module_available("torchvision")
|
|
DALI_AVAILABLE = _module_available("nvidia.dali")
|