-
Notifications
You must be signed in to change notification settings - Fork 101
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Compile Error due to a type mismatch #568
Comments
Ah yeah the gradient state needs to match the policy. I will look into this. In the mean time is there a reason you need f64 specifically? |
@seanmor5 Thanks 👍 Regarding f64, not really it is just an error that I came across while trying to hunt down the reason behind a NaN in one of the models. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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 the following policy
policy = Axon.MixedPrecision.create_policy( params: {:f, 64}, compute: {:f, 64}, output: {:f, 64} )
The error seems to be due to the
gradient_state
below initiated as:f32
axon/lib/axon/loop.ex
Line 360 in ddc49cc
The text was updated successfully, but these errors were encountered: