1

我是 Julia 的新手,由于某种原因,我无法让这个非常简单的代码工作。无论我尝试什么,我都会得到错误LoadError: Mutating arrays is not supported。我知道当我在优化过程中对数组进行变异以使代码不再可区分时,会发生此错误。我显然对 Julia 的了解还不够,无法看到我在做什么。

如果它有帮助,则错误似乎发生在该行中for d in dataset

using Statistics
using Flux: onehotbatch, onehot, onecold, crossentropy, throttle
using Flux
using Base.Iterators:repeated
using Plots:heatmap
using ImageView:imshow

images = Flux.Data.MNIST.images()[1:10]
labels = Flux.Data.MNIST.labels()[1:10]

heatmap(images[4], color=:grays, aspect_ratio=1)

X = float.(reshape.(images, :))
encode(x) = onehot(x, 0:9)
Y = encode.(labels)

m = Chain(Dense(28^2, 32, relu), Dense(32, 10), softmax)

loss(x, y) = crossentropy(m(x), y)
opt = ADAM()

accuracy(x, y) = mean(onecold(m(x)) .== onecold(y))

dataset = zip(X, Y)

print(size(X))

evalcb = () -> @show(loss(X, Y))

print("Training...")
# Flux.train!(loss, params(m), dataset, opt, cb=throttle(evalcb, 5));
for d in dataset
    print(d[2])
    gs = gradient(params(m)) do
        l = loss(d...)
    end
    update!(opt, params(m), gs)
end
4

1 回答 1

0

看起来我确实有一个旧版本的 Flux(但不是那么旧)。我必须卸载并重新安装 Julia 才能安装新版本的 Flux。

于 2021-04-23T22:01:46.850 回答