axon
axon copied to clipboard
Implement loops as Streams
This takes the regular loop run
and builds it based off of this stream
which will return a loop state per epoch. Basically this should be really flexible in that I can now control the running of the loop with regular Elixir Stream
and Enum
functions. You can see this in how much simpler the older run
is now.
This also kills output_transform
in the loop. I may add it to the Loop.run
as an option and that case we can add some output transforms that will not break some older training code
Resolves #362
Will potentially resolve #455 depending on how my experiments go