Skip to content

GitLab

"python/vscode:/vscode.git/clone" did not exist on "8154de5a326f945b514a98d075361db95eadd6ad"
Commit 24b8c1d1 authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal clean-up 

PiperOrigin-RevId: 402851292
parent 5072a869
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/modeling/layers/nn_layers.py
View file @ 24b8c1d1
...@@ -34,7 +34,7 @@ def make_divisible(value: float, ...@@ -34,7 +34,7 @@ def make_divisible(value: float,
Args: Args:
value: A `float` of original value. value: A `float` of original value.
divisor: An `int` off the divisor that need to be checked upon. divisor: An `int` of the divisor that need to be checked upon.
min_value: A `float` of minimum value threshold. min_value: A `float` of minimum value threshold.
Returns: Returns:
...@@ -46,7 +46,7 @@ def make_divisible(value: float, ...@@ -46,7 +46,7 @@ def make_divisible(value: float,
# Make sure that round down does not go down by more than 10%. # Make sure that round down does not go down by more than 10%.
if new_value < 0.9 * value: if new_value < 0.9 * value:
new_value += divisor new_value += divisor
return new_value return int(new_value)
def round_filters(filters: int, def round_filters(filters: int,
......
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