|
|
|
|
|
import os, sys
|
|
|
|
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
|
from pathlib import Path
|
|
|
import logging
|
|
|
|
|
|
import pytorch_lightning
|
|
|
|
|
|
|
|
|
formatter = logging.Formatter(
|
|
|
fmt="[%(asctime)s %(name)s %(levelname)s] %(message)s",
|
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
|
)
|
|
|
handler = logging.StreamHandler()
|
|
|
handler.setFormatter(formatter)
|
|
|
handler.setLevel(logging.INFO)
|
|
|
|
|
|
logger = logging.getLogger("mapper")
|
|
|
logger.setLevel(logging.INFO)
|
|
|
logger.addHandler(handler)
|
|
|
logger.propagate = False
|
|
|
|
|
|
pl_logger = logging.getLogger("pytorch_lightning")
|
|
|
if len(pl_logger.handlers):
|
|
|
pl_logger.handlers[0].setFormatter(formatter)
|
|
|
|
|
|
repo_dir = Path(__file__).parent.parent
|
|
|
EXPERIMENTS_PATH = repo_dir / "experiments/"
|
|
|
DATASETS_PATH = repo_dir / "datasets/"
|
|
|
|
