diff --git a/modules/GSOC/E2_ESRGAN/main.py b/modules/GSOC/E2_ESRGAN/main.py index 26abe90bbe07403fbb69680ae37778a4971a5a1e..456af1136ab89cf6814b853f07963afabd4c1c88 100644 --- a/modules/GSOC/E2_ESRGAN/main.py +++ b/modules/GSOC/E2_ESRGAN/main.py @@ -1,4 +1,5 @@ import os +from pathlib import Path from functools import partial import argparse # from absl import logging @@ -154,11 +155,11 @@ parser.add_argument( help="each 'v' increases vebosity of logging.") # FLAGS, unparsed = parser.parse_known_args() -FLAGS = {'config': 'config/config.yaml', +FLAGS = {'config': Path.home() + '/dev/python/modules/GSOC/E2_ESRGAN' + '/config/config.yaml', 'data_dir': None, 'manual': False, - 'model_dir': '/Users/tomrink/tf_model_esrgan/', - 'log_dir': '/Users/tomrink/tf_logs_esrgan/', + 'model_dir': Path.home() + '/tf_model_esrgan/', + 'log_dir': Path.home() + '/tf_logs_esrgan/', 'phases': 'phase1_phase2', 'export_only': False, 'tpu': '',