Fixes for Python 3

BERT/convert_params.py

@@ -137,7 +137,7 @@ def parse(init_checkpoint):
         else:
             print("ignored param: %s" % var_name)
 
-        if fluid_param_name is not '':
+        if fluid_param_name != '':
             tf_fluid_param_name_map[var_name] = fluid_param_name
             tf_param_name_shape_map[var_name] = var_shape
             fluid_param_name = ''

BERT/model/bert.py

@@ -109,7 +109,7 @@ class BertModel(object):
         emb_out = pre_process_layer(
             emb_out, 'nd', self._prepostprocess_dropout, name='pre_encoder')
 
-        if self._dtype is "float16":
+        if self._dtype == "float16":
             self_attn_mask = fluid.layers.cast(
                 x=self_attn_mask, dtype=self._dtype)
 

BERT/run_squad.py

@@ -17,11 +17,13 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+import argparse
+import collections
+import multiprocessing
 import os
 import time
-import argparse
+
 import numpy as np
-import collections
 
 import paddle
 import paddle.fluid as fluid

ERNIE/batching.py

@@ -19,6 +19,8 @@ from __future__ import print_function
 
 import numpy as np
 
+from six.moves import xrange
+
 
 def mask(batch_tokens,
          seg_labels,

ERNIE/finetune/classifier.py

@@ -22,6 +22,8 @@ import numpy as np
 
 import paddle.fluid as fluid
 
+from six.moves import xrange
+
 from model.ernie import ErnieModel
 
 

ERNIE/finetune/sequence_label.py

@@ -25,6 +25,8 @@ import multiprocessing
 import paddle
 import paddle.fluid as fluid
 
+from six.moves import xrange
+
 from model.ernie import ErnieModel
 
 def create_model(args,

ERNIE/model/ernie.py

@@ -109,7 +109,7 @@ class ErnieModel(object):
         emb_out = pre_process_layer(
             emb_out, 'nd', self._prepostprocess_dropout, name='pre_encoder')
 
-        if self._dtype is "float16":
+        if self._dtype == "float16":
             self_attn_mask = fluid.layers.cast(
                 x=self_attn_mask, dtype=self._dtype)