fix issue 120
This commit is contained in:
Родитель
fc1cdcf682
Коммит
d7bbd25b80
|
@ -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 = ceil(in_hw // 2);
|
||||
if s = 4, out_hw = ceil(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)
|
||||
|
|
Загрузка…
Ссылка в новой задаче