Flux Model Zoo 所给Conv模型在运行时出错,更改错误代码后出现新错误,请问怎么解决
代码网站:https://github.com/FluxML/model-zoo/blob/master/vision/mnist/conv.jl
x - > maxpool(x, (2, 2))
改成MaxPool((2,2))
应该就好了
Conv((3, 3), 1=>16, pad=(1,1), relu),
x - > maxpool(x, (2, 2)),
改成
Conv((3, 3), 1=>16, pad=(1,1), relu),
Maxpool((2, 2)),
我恰好前两天跑了一下这个网络应该是没有问题的。。。
# Flux 0.8.3
model = Chain(
Conv((3, 3), 1=>16, pad=(1,1), relu),
MaxPool((2,2)),
Conv((3, 3), 16=>32, pad=(1,1), relu),
MaxPool((2,2)),
Conv((3, 3), 32=>32, pad=(1,1), relu),
MaxPool((2,2)),
x -> reshape(x, :, size(x, 4)),
Dense(288, 10),
softmax,
)