I’m trying to use CuIterator and Flux.train!. Attempting to do so gives an error because train! tries to get the length of the CuIterator. So this: iter = CuIterator(batch(i) for i in 1:BATCH_COUNT) Flux.train!(loss, ps, iter, opt) results in an error: MethodError: no method matching length(::CuI…

Error using CuIterator with Flux.train!

DrChainsaw August 27, 2022, 5:31pm 2

I have seen the same but forgot to make an issue/pr. I think the correct(er) fix is to delegate both HasLength and length to the wrapped iterator.

Topic		Replies	Views
What's wrong with this Flux model definitin? Machine Learning first-steps , flux	1	768	November 7, 2019
Flux.train failing with CUDA GPU Machine Learning	5	933	March 27, 2019
`CUDA error: out of memory` with Flux Machine Learning flux	4	1668	August 24, 2020
MNIST GPU CuArrays error GPU	23	3162	January 22, 2019
Error when setting up Flux and CuArrays Machine Learning gpu , flux	4	679	October 2, 2019

Error using CuIterator with Flux.train!

Related topics