提交 94a23780 authored 作者: Taesup (TS) Kim's avatar Taesup (TS) Kim

issue 4303 : fix reshaping part to deal with batch_size==None and

num_input_channels==None
上级 7943ab37
...@@ -567,7 +567,7 @@ def bilinear_upsampling(input, ...@@ -567,7 +567,7 @@ def bilinear_upsampling(input,
subsample=(ratio, ratio), subsample=(ratio, ratio),
filter_flip=True) filter_flip=True)
return upsampled_mat.reshape((batch_size, num_input_channels, return upsampled_mat.reshape((input.shape[0], input.shape[1],
row * ratio, col * ratio)) row * ratio, col * ratio))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论