Flux文档conv模型运行时出现错误


#1
Flux Model Zoo 所给Conv模型在运行时出错,更改错误代码后出现新错误,请问怎么解决
代码网站:https://github.com/FluxML/model-zoo/blob/master/vision/mnist/conv.jl

image

image


#2

x - > maxpool(x, (2, 2))改成MaxPool((2,2))应该就好了


#3

改完之后会出现新的错误
image


#4
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)),

#5

好像还是有错误,官方文档复制出来的代码直接运行就会出错。总之谢谢了
image


#6

我恰好前两天跑了一下这个网络:man_facepalming:应该是没有问题的。。。

# 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,
)