Merge pull request #16 from thongonary/master

python3 compatible

MPIDriver.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 ### This script creates an MPIManager object and launches distributed training.
 

mpi_learn/mpi/process.py

@@ -118,7 +118,7 @@ def print_metrics(self, metrics):
         """Display metrics computed during training or validation"""
         names = self.model.metrics_names
         if len(names) == 1:
-            print "%s: %.3f" % (names[0],metrics)
+            print ("%s: %.3f" % (names[0],metrics))
         else:
             for name, metric in zip( names, metrics ):
                 print ("{0}: {1:.3f}".format(name,metric))

mpi_learn/train/model.py

@@ -31,7 +31,7 @@ def __init__(self, comm, filename=None,json_str=None, custom_objects={}, weights
         self.filename = filename
         self.json_str = json_str
         self.weights = weights
-	self.custom_objects = custom_objects
+        self.custom_objects = custom_objects
         super(ModelFromJson, self).__init__(comm)
 
     def build_model(self):
@@ -51,7 +51,7 @@ def __init__(self, comm, filename=None, json_str=None, device_name='cpu',
         self.filename = filename
         self.json_str = json_str
         self.weights = weights
-	self.custom_objects = custom_objects
+        self.custom_objects = custom_objects
         self.device = self.get_device_name(device_name)
         super(ModelFromJsonTF, self).__init__(comm)
 
@@ -66,11 +66,11 @@ def get_device_name(self, device):
                 dev_num = int(device[3:])
                 dev_type = 'gpu'
             except ValueError:
-                print "GPU number could not be parsed from {}; using CPU".format(device)
+                print ("GPU number could not be parsed from {}; using CPU".format(device))
                 dev_num = 0
                 dev_type = 'cpu'
         else:
-            print "Please specify 'cpu' or 'gpuN' for device name"
+            print ("Please specify 'cpu' or 'gpuN' for device name")
             dev_num = 0
             dev_type = 'cpu'
         return get_device_name(dev_type, dev_num, backend='tensorflow')