Add pytorch version reporting

2020-09-13 16:14:48 -04:00 · 2020-09-13 16:14:48 -04:00 · b7a9adf0c7
commit b7a9adf0c7
parent a579c9655b
2 changed files with 2 additions and 0 deletions
--- a/map2map/test.py
+++ b/map2map/test.py
@ -11,6 +11,7 @@ from .utils import import_attr, load_model_state_dict


 def test(args):
+    print('pytorch {}'.format(torch.__version__))
    pprint(vars(args))
    sys.stdout.flush()

--- a/map2map/train.py
+++ b/map2map/train.py
@ -184,6 +184,7 @@ def gpu_worker(local_rank, node, args):
        logger = SummaryWriter()

    if rank == 0:
+        print('pytorch {}'.format(torch.__version__))
        pprint(vars(args))
        sys.stdout.flush()