提交 b93cc86d authored 作者: Guillaume Alain's avatar Guillaume Alain 提交者: Arnaud Bergeron

Fixed some typos and a dimension that was probably wrong. This doesn't solve the…

Fixed some typos and a dimension that was probably wrong. This doesn't solve the issue with the boder_mode='full' because we still have other errors.
上级 dbe9c96a
...@@ -382,7 +382,7 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None, ...@@ -382,7 +382,7 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None,
input: (b, ic, i0, i1) input: (b, ic, i0, i1)
filters: (oc, ic, f0, f1) filters: (oc, ic, f0, f1)
bocder_mode: 'valid' of 'full' border_mode: 'valid' of 'full'
""" """
# use symbolic shapes to compute shape info at runtime if not specified # use symbolic shapes to compute shape info at runtime if not specified
...@@ -405,13 +405,13 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None, ...@@ -405,13 +405,13 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None,
filters_padded = T.set_subtensor(filters_padded[:, :, :f0, :f1], filters_padded = T.set_subtensor(filters_padded[:, :, :f0, :f1],
filters) filters)
input_padded = input input_padded = input
elif mode == 'full': elif border_mode == 'full':
o0 = i0 + f0 - 1 o0 = i0 + f0 - 1
o1 = i1 + f1 - 1 o1 = i1 + f1 - 1
filters_padded = T.zeros((oc, ic, o0, o1), dtype='float32') filters_padded = T.zeros((oc, ic, o0, o1), dtype='float32')
filters_padded = T.set_subtensor(filters_padded[:, :, :f0, :f1], filters_padded = T.set_subtensor(filters_padded[:, :, :f0, :f1],
filters) filters)
input_padded = T.zeros((oc, ic, o0, o1), dtype='float32') input_padded = T.zeros((b, ic, o0, o1), dtype='float32')
input_padded = T.set_subtensor(input_padded[:, :, :i0, :i1], input_padded = T.set_subtensor(input_padded[:, :, :i0, :i1],
input) input)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论