提交 16d8b099 authored 作者: Frederic Bastien's avatar Frederic Bastien

Small fix to a test

上级 824cb369
...@@ -1505,7 +1505,7 @@ def test_dnn_batchnorm_train_without_running_averages(): ...@@ -1505,7 +1505,7 @@ def test_dnn_batchnorm_train_without_running_averages():
bn.batch_normalization_train(x, scale, bias, 'per-activation') bn.batch_normalization_train(x, scale, bias, 'per-activation')
# backward pass # backward pass
grads_gpu = T.grad(None, wrt=[x, scale, bias], known_grads={out_gpu: dy}) grads_gpu = T.grad(None, wrt=[x, scale, bias], known_grads={out_gpu: dy})
grads_abstract = T.grad(None, wrt=[x, scale, bias], known_grads={out_gpu: dy}) grads_abstract = T.grad(None, wrt=[x, scale, bias], known_grads={out_abstract: dy})
# compile # compile
f_gpu = theano.function([x, scale, bias, dy], f_gpu = theano.function([x, scale, bias, dy],
[out_gpu, x_mean_gpu, x_invstd_gpu] + [out_gpu, x_mean_gpu, x_invstd_gpu] +
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论