maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix the LGTM alert

Browse source
This commit is contained in:
BO-YANG HSUEH 2020-03-06 19:37:14 +08:00 committed by GitHub
parent ecfad37240
commit 4b421a9656
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
FasterTransformer/v2/sample/tensorflow/translate_sample.py
View file

@ -15,17 +15,11 @@
from __future__ import print_function
import tensorflow as tf
import numpy as np
import os
import math
import six
import argparse
from utils.common import time_test, DecodingArgument, int_result_cross_check, TransformerArgument
from utils.common import time_test, DecodingArgument
from utils.decoding import tf_decoding, generate_encoder_result, op_decoding
from utils.encoder import tf_encoder, op_encoder
from utils.decoder import build_sequence_mask
from opennmt.utils import misc
from opennmt.utils import compat
from opennmt.encoders.self_attention_encoder import SelfAttentionEncoder
from opennmt.decoders.self_attention_decoder import SelfAttentionDecoder
from opennmt.inputters import WordEmbedder
@ -81,7 +75,6 @@ if __name__ == "__main__":
np.random.seed(1)
tf.set_random_seed(1)
kernel_initializer_range = 0.02
initializer_range = 0.02
bias_initializer_range = 0.02
batch_size = args.batch_size