fix py2/py3 compatity (PaddlePaddle#340)

lizexu123 · Jun 9, 2020 · 2ff4352 · 2ff4352
1 parent e028149
commit 2ff4352
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/paddleslim/nas/darts/search_space/conv_bert/reader/cls.py b/paddleslim/nas/darts/search_space/conv_bert/reader/cls.py
@@ -17,8 +17,8 @@
 import types
 import csv
 import numpy as np
-import tokenization
-from batching import prepare_batch_data
+from . import tokenization
+from .batching import prepare_batch_data
 
 
 class DataProcessor(object):

diff --git a/paddleslim/nas/darts/train_search.py b/paddleslim/nas/darts/train_search.py
@@ -20,7 +20,6 @@
 
 import os
 import logging
-from itertools import izip
 import numpy as np
 import paddle.fluid as fluid
 from paddle.fluid.dygraph.base import to_variable
@@ -107,7 +106,7 @@ def train_one_epoch(self, train_loader, valid_loader, architect, optimizer,
         self.model.train()
 
         step_id = 0
-        for train_data, valid_data in izip(train_loader(), valid_loader()):
+        for train_data, valid_data in zip(train_loader(), valid_loader()):
             if epoch >= self.epochs_no_archopt:
                 architect.step(train_data, valid_data)
 

diff --git a/paddleslim/teachers/bert/reader/cls.py b/paddleslim/teachers/bert/reader/cls.py
@@ -17,8 +17,8 @@
 import types
 import csv
 import numpy as np
-import tokenization
-from batching import prepare_batch_data
+from . import tokenization
+from .batching import prepare_batch_data
 
 
 class DataProcessor(object):