mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
34 lines
709 B
Python
34 lines
709 B
Python
""" Layer/Module Helpers
|
|
|
|
Hacked together by Ross Wightman
|
|
"""
|
|
from itertools import repeat
|
|
from torch._six import container_abcs
|
|
|
|
|
|
# From PyTorch internals
|
|
def _ntuple(n):
|
|
def parse(x):
|
|
if isinstance(x, container_abcs.Iterable):
|
|
return x
|
|
return tuple(repeat(x, n))
|
|
return parse
|
|
|
|
|
|
tup_single = _ntuple(1)
|
|
tup_pair = _ntuple(2)
|
|
tup_triple = _ntuple(3)
|
|
tup_quadruple = _ntuple(4)
|
|
|
|
|
|
def make_divisible(v, divisor=8, min_value=None):
|
|
min_value = min_value or divisor
|
|
new_v = max(min_value, int(v + divisor / 2) // divisor * divisor)
|
|
# Make sure that round down does not go down by more than 10%.
|
|
if new_v < 0.9 * v:
|
|
new_v += divisor
|
|
return new_v
|
|
|
|
|
|
|