in run_model.py [0:0]
def setup_logger(config):
# Removing handlers that might be associated with environment; and logs
# out to both stderr and a log file
for handler in logging.root.handlers[:]:
logging.root.removeHandler(handler)
log_file_name = os.path.join(config.get("EXPERIMENT", "experiment_directory"), "experiment.log")
logging.basicConfig(
format='%(asctime)s [%(levelname)s] %(message)s',
datefmt='%m/%d/%Y %I:%M:%S %p',
level=logging.DEBUG,
handlers=[
logging.FileHandler(log_file_name),
logging.StreamHandler()
]
)
logging.info(f"Initializing experiment: {config.get('EXPERIMENT', 'experiment_directory')}")
logging.info(f"Running model on device: {device}")