提交 80e32e42 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Fixed crash due to integer division in Conv3D

上级 1c07a240
...@@ -135,9 +135,9 @@ class Conv3D(theano.Op): ...@@ -135,9 +135,9 @@ class Conv3D(theano.Op):
vidDur = V_shape[3] vidDur = V_shape[3]
filterDur = W_shape[3] filterDur = W_shape[3]
output_height = T.floor( (vidHeight - filterHeight) / dr )+1 output_height = T.floor((vidHeight - filterHeight) // dr) + 1
output_width = T.floor( (vidWidth - filterWidth) / dc )+1 output_width = T.floor((vidWidth - filterWidth) // dc) + 1
output_dur = T.floor( (vidDur - filterDur) / dt ) +1 output_dur = T.floor((vidDur - filterDur) // dt) + 1
rval = (batch_size, output_height, output_width, output_dur, output_channels ) rval = (batch_size, output_height, output_width, output_dur, output_channels )
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论