SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

remove ipdb

Browse Source
This commit is contained in:
Patrick von Platen
2020-03-09 20:05:41 +01:00
parent 374deef48d
commit cf06290565
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/modeling_tf_utils.py
View File

@@ -641,9 +641,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
# create attention mask if necessary # create attention mask if necessary
# TODO (PVP): this should later be handled by the forward fn() in each model in the future see PR 3140 # TODO (PVP): this should later be handled by the forward fn() in each model in the future see PR 3140
import ipdb
ipdb.set_trace()
if (attention_mask is None) and (pad_token_id is not None) and (pad_token_id in input_ids.numpy()): if (attention_mask is None) and (pad_token_id is not None) and (pad_token_id in input_ids.numpy()):
attention_mask = tf.cast(tf.math.not_equal(input_ids, pad_token_id), dtype=tf.int32) attention_mask = tf.cast(tf.math.not_equal(input_ids, pad_token_id), dtype=tf.int32)
elif attention_mask is None: elif attention_mask is None: