diff --git a/eval.py b/eval.py index 43148c7..0c92eb5 100644 --- a/eval.py +++ b/eval.py @@ -17,6 +17,8 @@ from metrics import Metrics from checkpoint import load_checkpoint +import pprint + def eval(**args): """ Evaluate selected model @@ -33,9 +35,8 @@ def eval(**args): None """ - print("\n############################################################################\n") - print("Experimental Setup: ", args) - print("\n############################################################################\n") + print("Experimental Setup: ") + pprint.PrettyPrinter(indent=4).pprint(args) d = datetime.datetime.today() date = d.strftime('%Y%m%d-%H%M%S') diff --git a/train.py b/train.py index d889d84..6769c63 100644 --- a/train.py +++ b/train.py @@ -18,6 +18,8 @@ from metrics import Metrics from checkpoint import save_checkpoint, load_checkpoint +import pprint + def train(**args): """ Evaluate selected model @@ -41,9 +43,8 @@ def train(**args): None """ - print("\n############################################################################\n") - print("Experimental Setup: ", args) - print("\n############################################################################\n") + print("Experimental Setup: ") + pprint.PrettyPrinter(indent=4).pprint(args) for total_iteration in range(args['rerun']):