pytorch version logging added

bioinform · Mar 1, 2019 · df51e9d · df51e9d
1 parent e77c242
commit df51e9d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/neusomatic/python/call.py b/neusomatic/python/call.py
@@ -20,6 +20,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 from torchvision import transforms
+import torchvision
 
 from network import NeuSomaticNet
 from dataloader import NeuSomaticDataset
@@ -391,6 +392,8 @@ def call_neusomatic(candidates_tsv, ref_file, out_dir, checkpoint, num_threads,
 
     logger.info("-----------------Call Somatic Mutations--------------------")
 
+    logger.info("PyTorch Version: {}".format(torch.__version__))
+    logger.info("Torchvision Version: {}".format(torchvision.__version__))
     if not use_cuda:
         torch.set_num_threads(num_threads)
 

diff --git a/neusomatic/python/train.py b/neusomatic/python/train.py
@@ -16,6 +16,7 @@
 import torch.nn.functional as F
 import torch.optim as optim
 from torchvision import transforms
+import torchvision
 from random import shuffle
 import pickle
 
@@ -191,6 +192,8 @@ def train_neusomatic(candidates_tsv, validation_candidates_tsv, out_dir, checkpo
     logger = logging.getLogger(train_neusomatic.__name__)
 
     logger.info("----------------Train NeuSomatic Network-------------------")
+    logger.info("PyTorch Version: {}".format(torch.__version__))
+    logger.info("Torchvision Version: {}".format(torchvision.__version__))
 
     if not os.path.exists(out_dir):
         os.mkdir(out_dir)