提交 71000876 authored 作者: erakra's avatar erakra

fixing input type

上级 b312b4de
......@@ -1838,7 +1838,7 @@ def frac_bilinear_upsampling(input,
pad = double_pad // 2
# build pyramidal kernel
kern = bilinear_kernel_2D(ratio=ratio)[np.newaxis, np.newaxis, :, :]
kern = bilinear_kernel_2D(ratio=ratio)[np.newaxis, np.newaxis, :, :].astype(theano.config.floatX)
# add corresponding padding
pad_kern = T.concatenate((T.zeros(tuple(kern.shape[:2]) + (pad[0], kern.shape[-1]),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论