Skip to content
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 fine tuning #128

Merged
merged 12 commits into from
May 29, 2024
Merged

add fine tuning #128

merged 12 commits into from
May 29, 2024

Conversation

federicazanca
Copy link
Collaborator

@federicazanca federicazanca commented May 23, 2024

Closes #127
--fine_tune key added as an input for the train Calcjob
The input validation checks that a model is given either in the config file or as an input.
in the config file, it can be anything (a string referring to some model types like "small" or a path to a model file)
If it is given as a ModelData input, it has to be a file.
This is in general a main difference between janus and aiida-mlip that I don't know if we want to keep like this or change (the problem being that ModelData is built to be a file, so if we want it to also be a string, I can either completely modify it or just allow an additional input that is an alternative to the ModelData and it is just a string)

  • check that there aren't other checks to do
  • more tests could be added where I actually run the calculation; I just wasn't sure because of the mace development thing
  • docs

@federicazanca federicazanca added enhancement New/improved feature or request aiida-mlip labels May 23, 2024
@federicazanca federicazanca self-assigned this May 23, 2024
oerc0122
oerc0122 previously approved these changes May 23, 2024
Copy link
Collaborator

@oerc0122 oerc0122 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good, just optional stylistic things

aiida_mlip/calculations/train.py Outdated Show resolved Hide resolved
tests/calculations/test_fine-tuning.py Outdated Show resolved Hide resolved
Copy link
Member

@ElliottKasoar ElliottKasoar left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just a couple of questions/checks but looks good!

aiida_mlip/calculations/train.py Show resolved Hide resolved
aiida_mlip/calculations/train.py Outdated Show resolved Hide resolved
aiida_mlip/calculations/train.py Show resolved Hide resolved
tests/calculations/test_train.py Outdated Show resolved Hide resolved
tests/calculations/test_train.py Outdated Show resolved Hide resolved
@federicazanca federicazanca marked this pull request as ready for review May 28, 2024 12:59
pyproject.toml Outdated Show resolved Hide resolved
pyproject.toml Outdated Show resolved Hide resolved
Co-authored-by: Alin Marin Elena <[email protected]>
@alinelena alinelena self-requested a review May 28, 2024 18:09
@alinelena alinelena self-requested a review May 28, 2024 18:10
alinelena
alinelena previously approved these changes May 28, 2024
oerc0122
oerc0122 previously approved these changes May 29, 2024
@alinelena alinelena merged commit 0fdacf2 into stfc:main May 29, 2024
8 checks passed
@federicazanca federicazanca deleted the 127-add-fine-tuning branch May 29, 2024 16:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New/improved feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

add fine tuning
4 participants