axon
axon copied to clipboard
Implement loops as Streams
trafficstars
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