This commit is contained in:
jiahangxu 2023-05-04 05:09:16 -04:00
Родитель 29bfef7d83
Коммит fc1cdcf682
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -17,8 +17,8 @@ def get_inputs_by_shapes(shapes, batch_size = 1):
def get_padding(ks, s, hw):
""" choose padding value to make sure:
if s = 1, out_hw = in_hw;
if s = 2, out_hw = in_hw // 2;
if s = 4, out_hw = in_hw // 4;
if s = 2, out_hw = ceil(in_hw // 2);
if s = 4, out_hw = ceil(in_hw // 4);
"""
if hw % s == 0:
pad = max(ks - s, 0)