Description
Hi,
I tried the tutorial on this repo that is mentioned in the title. I am using Julia v. 1.8.5, and
[052768ef] CUDA v4.0.1
[587475ba] Flux v0.13.12
[eb30cadb] MLDatasets v0.7.9
ERROR: LoadError: Scalar indexing is disallowed.
Invocation of getindex resulted in scalar indexing of a GPU array.
This is typically caused by calling an iterating implementation of a method.
Such implementations do not execute on the GPU, but very slowly on the CPU
According to the line numbers (in my file script), the issue seems to come from:
for i in 1:10
preds = m(test[i][1])
lab = test[i][2]
for j = 1:1000
pred_class = findmax(preds[:, j])[2]
actual_class = findmax(lab[:, j])[2]
if pred_class == actual_class
class_correct[pred_class] += 1
end
class_total[actual_class] += 1
end
end
Besides this, I saw the warning:
┌ Warning: CIFAR10.testdata() is deprecated, use CIFAR10(split=:test)[:]
instead.
/Pedro