Skip to content

GitLab

Commit 8d11ee24 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

internal change 

PiperOrigin-RevId: 333014322
parent c3d53565
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/nlp/configs/encoders.py
View file @ 8d11ee24
......@@ -203,6 +203,7 @@ def build_encoder(
hidden_dropout_prob=encoder_cfg.hidden_dropout_prob,
attention_probs_dropout_prob=encoder_cfg.attention_probs_dropout_prob,
intra_bottleneck_size=encoder_cfg.intra_bottleneck_size,
initializer_range=encoder_cfg.initializer_range,
key_query_shared_bottleneck=encoder_cfg.key_query_shared_bottleneck,
num_feedforward_networks=encoder_cfg.num_feedforward_networks,
normalization_type=encoder_cfg.normalization_type,
......
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