import sys
import logging
import time
from data import build_loaders
from util.config import Config
from torch.utils.tensorboard import SummaryWriter
from models import get_trainer
torch.backends.cudnn.benchnark = True
# python train inpaint.yml
args = Config(sys.argv[2])
logger = logging.getLogger(__name__)
time_stamp = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
log_dir = 'model_logs/{}/{}_{}'.format(args.trainer, time_stamp, args.log_dir)
args.log_id = os.path.join(time_stamp, args.log_dir)
result_dir = 'result_logs/{}/{}_{}'.format(args.trainer, time_stamp,
args.log_dir)
tensorboard_logger = SummaryWriter(log_dir)
if not os.path.exists(log_dir):
os.makedirs(log_dir)
if not os.path.exists(result_dir):
os.makedirs(result_dir)
torch.backends.cudnn.benchmark = True
def logger_init():
"""
Initialize the logger to some file.
"""
