diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index 94414237dda5cf0334c59492720b0fc403d3c376..22fef508582a5c0822c13be958c163f556a1d2d6 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/modules/GSOC/E2_ESRGAN/lib/train.py @@ -7,8 +7,9 @@ import tensorflow as tf # from lib import utils, dataset from GSOC.E2_ESRGAN.lib import utils, dataset import glob +from pathlib import Path -logging.basicConfig(filename='/Users/tomrink/esrgan_log.txt', level=logging.DEBUG) +logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG) class Trainer(object):