riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-06 06:10:04 +06:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

whitespace

Browse Source
This commit is contained in:
Julien Chaumond 2019-11-11 15:18:09 -05:00
parent d409aca326
commit dd6b2e05e1
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -137,4 +137,5 @@ examples/runs
serialization_dir serialization_dir
# emacs # emacs
*.*~ *.*~
debug.env

14
transformers/modeling_tf_utils.py
View File

@ -477,10 +477,10 @@ def shape_list(x):
return [dynamic[i] if s is None else s for i, s in enumerate(static)] return [dynamic[i] if s is None else s for i, s in enumerate(static)]
def get_initializer(initializer_range=0.02): def get_initializer(initializer_range=0.02):
"""Creates a `tf.initializers.truncated_normal` with the given range. """Creates a `tf.initializers.truncated_normal` with the given range.
Args: Args:
initializer_range: float, initializer range for stddev. initializer_range: float, initializer range for stddev.
Returns: Returns:
TruncatedNormal initializer with stddev = `initializer_range`. TruncatedNormal initializer with stddev = `initializer_range`.
""" """
return tf.keras.initializers.TruncatedNormal(stddev=initializer_range) return tf.keras.initializers.TruncatedNormal(stddev=initializer_range)