axon

axon copied to clipboard

seanmor5

seanmor5

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...

seanmor5

Compile Error due to a type mismatch

2

I am getting the following error `(CompileError) deps/axon/lib/axon/loop.ex:469: the do-block in while must return tensors with the same shape, type, and names as the initial arguments.` While trying to enforce...

Mostafa86