From b7a9adf0c7dfe27e47949fada0cb59d0c58e7563 Mon Sep 17 00:00:00 2001
From: Yin Li <eelregit@gmail.com>
Date: Sun, 13 Sep 2020 16:14:48 -0400
Subject: [PATCH] Add pytorch version reporting

---
 map2map/test.py  | 1 +
 map2map/train.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/map2map/test.py b/map2map/test.py
index dc87486..d2b86a1 100644
--- 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()
 
diff --git a/map2map/train.py b/map2map/train.py
index 0898563..091364a 100644
--- 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()