I have been playing with the model zoo autoencoder code. I have a question about this line of code:
@epochs 10 Flux.train!(loss, params(m), zip(data), opt, cb = evalcb)
zip(data) doing to the data, and what type does
Flux::train! expects for the data argument?
I am confused because data is an array of 60 batches, and zip as only the one argument data. If I pass data directly, it stalls and does nothing. I suspect it has to do with the fact that the type of
zip(data) is iterator, and may-be that is what
Flux::train! needs for efficiency, but since I can’t find documentation for
train!, I have no clue.
Any hints appreciated. Thanks!