Skip to content

GitLab

"docs/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "3549c57bc3745cb4e372a526386bcd1b69fc2d61"
Commit e4bf28fc authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

fix lint errors

parent 2a56bb7e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
official/transformer/v2/transformer_main.py
View file @ e4bf28fc
...@@ -99,10 +99,11 @@ class TransformerTask(object): ...@@ -99,10 +99,11 @@ class TransformerTask(object):
print("Running transformer with num_gpus =", num_gpus) print("Running transformer with num_gpus =", num_gpus)
if self.distribution_strategy: if self.distribution_strategy:
print("For training, using distribution strategy: ", self.distribution_strategy) print("For training, using distribution strategy: ",
self.distribution_strategy)
else: else:
print("Not using any distribution strategy.") print("Not using any distribution strategy.")
self.params = params = misc.get_model_params(flags_obj.param_set, num_gpus) self.params = params = misc.get_model_params(flags_obj.param_set, num_gpus)
params["num_gpus"] = num_gpus params["num_gpus"] = num_gpus
...@@ -197,7 +198,7 @@ class TransformerTask(object): ...@@ -197,7 +198,7 @@ class TransformerTask(object):
with tf.name_scope("model"): with tf.name_scope("model"):
model = transformer.create_model(params, is_train) model = transformer.create_model(params, is_train)
self._load_weights_if_possible( self._load_weights_if_possible(
model, tf.train.latest_checkpoint(self.flags_obj.model_dir)) model, tf.train.latest_checkpoint(self.flags_obj.model_dir))
model.summary() model.summary()
subtokenizer = tokenizer.Subtokenizer(flags_obj.vocab_file) subtokenizer = tokenizer.Subtokenizer(flags_obj.vocab_file)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment