axon icon indicating copy to clipboard operation
axon copied to clipboard

Implement loops as Streams

Open seanmor5 opened this issue 2 months ago • 0 comments

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

seanmor5 avatar May 11 '24 18:05 seanmor5