-
Notifications
You must be signed in to change notification settings - Fork 3
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
Add multi-precision support #23
base: main
Are you sure you want to change the base?
Conversation
First version should 0.1.0
…s/fix-version Version 0.1.0
…s/PR-Codecov Pr codecov
* Cleaning the Todos --------- Co-authored-by: tmigot <[email protected]>
Co-authored-by: tmigot <[email protected]>
This reverts commit f92d0db.
fixing a typo in the readme
…s/farhadrclass-patch-1 Update README.md
|
||
Tests FluxNLPModel loader and NN device consistency. | ||
""" | ||
function test_devices_consistency(chain_ANN::Vector{T},data_train,data_test) where {T <: Chain} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
why do we need this? loader and training can be on different machine or type, we just need to adjust them on the running
@@ -0,0 +1,78 @@ | |||
# test example taken from Flux quickstart guide ([https://fluxml.ai/Flux.jl/stable/models/quickstart/](https://fluxml.ai/Flux.jl/stable/models/quickstart/)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I suggest having couple of more tests to include one or 2 iteration with training and updating the weights
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we can add this to my PR, I will have it ready soon but the tests look good
No description provided.