Merge pull request #367 from wuyefeilin/dygraph

update train.py and get_environ_info.py

dygraph/core/train.py

@@ -79,7 +79,7 @@ def train(model,
     train_batch_cost = 0.0
     timer.start()
 
-    iter = 0
+    iter = start_iter
     while iter < iters:
         for data in loader:
             iter += 1

dygraph/models/unet.py

@@ -64,7 +64,6 @@ class UNet(fluid.dygraph.Layer):
         """
         if pretrained_model is not None:
             if os.path.exists(pretrained_model):
-                utils.load_pretrained_model(self.backbone, pretrained_model)
                 utils.load_pretrained_model(self, pretrained_model)
             else:
                 raise Exception('Pretrained model is not found: {}'.format(

dygraph/utils/get_environ_info.py

@@ -83,11 +83,16 @@ def get_environ_info():
     env_info = {}
     env_info['System Platform'] = sys.platform
     if env_info['System Platform'] == 'linux':
-        lsb_v = subprocess.check_output(['lsb_release', '-v']).decode().strip()
+        try:
+            lsb_v = subprocess.check_output(['lsb_release',
+                                             '-v']).decode().strip()
             lsb_v = lsb_v.replace('\t', ' ')
-        lsb_d = subprocess.check_output(['lsb_release', '-d']).decode().strip()
+            lsb_d = subprocess.check_output(['lsb_release',
+                                             '-d']).decode().strip()
             lsb_d = lsb_d.replace('\t', ' ')
             env_info['LSB'] = [lsb_v, lsb_d]
+        except:
+            pass
 
     env_info['Python'] = sys.version.replace('\n', '')