Skip to content

ENH add MixedPrecision + improve compile#7

Merged
tomMoral merged 3 commits intomainfrom
ENH_improving_optimizers
Oct 31, 2025
Merged

ENH add MixedPrecision + improve compile#7
tomMoral merged 3 commits intomainfrom
ENH_improving_optimizers

Conversation

@tomMoral
Copy link
Member

@tomMoral tomMoral commented Oct 30, 2025

This provides a clear speed up! @tonysf
Note that now, 6000 iterations takes 20min
image

There seems to have a weird behavior for Scion when changing the learning rate decay....

@tomMoral tomMoral mentioned this pull request Oct 30, 2025
8 tasks
@tomMoral
Copy link
Member Author

I fixed the issue with the recompilation of a step in the middle of the training, now the iteration time is coherent:
image

I will merge this one as it already improves the training a lot.

@tomMoral tomMoral merged commit f6246f9 into main Oct 31, 2025
4 checks passed
@tomMoral tomMoral deleted the ENH_improving_optimizers branch October 31, 2025 13:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant