diff --git a/modules/GSOC/E2_ESRGAN/main.py b/modules/GSOC/E2_ESRGAN/main.py index a65ea43e50315902c9702f32c08a6a3a7e429d46..aafd009fab1e03531af4f45b0f3f32067fed6e6c 100644 --- a/modules/GSOC/E2_ESRGAN/main.py +++ b/modules/GSOC/E2_ESRGAN/main.py @@ -149,7 +149,8 @@ parser.add_argument( FLAGS = {'config': str(Path.home()) + '/dev/python/modules/GSOC/E2_ESRGAN' + '/config/config.yaml', 'data_dir': None, 'manual': False, - 'model_dir': str(Path.home()) + '/tf_model_esrgan/', + #'model_dir': str(Path.home()) + '/tf_model_esrgan/', + 'model_dir': '/scratch/long/rink' + '/tf_model_esrgan/', 'log_dir': str(Path.home()) + '/tf_logs_esrgan/', 'phases': 'phase1_phase2', 'export_only': False,