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):
|
def get_padding(ks, s, hw):
|
||||||
""" choose padding value to make sure:
|
""" choose padding value to make sure:
|
||||||
if s = 1, out_hw = in_hw;
|
if s = 1, out_hw = in_hw;
|
||||||
if s = 2, out_hw = ceil(in_hw // 2);
|
if s = 2, out_hw = ceil(in_hw / 2);
|
||||||
if s = 4, out_hw = ceil(in_hw // 4);
|
if s = 4, out_hw = ceil(in_hw / 4);
|
||||||
"""
|
"""
|
||||||
if hw % s == 0:
|
if hw % s == 0:
|
||||||
pad = max(ks - s, 0)
|
pad = max(ks - s, 0)
|
||||||
|
|
Загрузка…
Ссылка в новой задаче