diff --git a/dygraph/core/train.py b/dygraph/core/train.py index 2bce9a16653c978024f8d5bb46ebc42459003dbb..6c64532fffc0951e79596cea17971b8d51e534f6 100644 --- a/dygraph/core/train.py +++ b/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 diff --git a/dygraph/models/unet.py b/dygraph/models/unet.py index 7a2b80bd939805fe8d447e56f42424eb036b06f3..4ce28400c8f7dd64f40d651506d13988aa764c39 100644 --- a/dygraph/models/unet.py +++ b/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( diff --git a/dygraph/utils/get_environ_info.py b/dygraph/utils/get_environ_info.py index 944c1f9595d5a4733df85f0b24b380bd00f8a726..7d789f4d60e875fd11514fa13b901885be7b0024 100644 --- a/dygraph/utils/get_environ_info.py +++ b/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() - lsb_v = lsb_v.replace('\t', ' ') - lsb_d = subprocess.check_output(['lsb_release', '-d']).decode().strip() - lsb_d = lsb_d.replace('\t', ' ') - env_info['LSB'] = [lsb_v, lsb_d] + 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 = lsb_d.replace('\t', ' ') + env_info['LSB'] = [lsb_v, lsb_d] + except: + pass env_info['Python'] = sys.version.replace('\n', '')