Skip to content

Add SevenNet to train#662

Open
harveydevereux wants to merge 21 commits intostfc:mainfrom
harveydevereux:add_sevennet_to_train
Open

Add SevenNet to train#662
harveydevereux wants to merge 21 commits intostfc:mainfrom
harveydevereux:add_sevennet_to_train

Conversation

@harveydevereux
Copy link
Collaborator

@harveydevereux harveydevereux commented Feb 3, 2026

  • train support for SevenNet```
  • fine-tuning
  • docs
  • test train
  • test fine-tune

The only curve ball with this one is that the atom attributes forces, energy, and stress must be called that, as far as I can tell.

@ElliottKasoar ElliottKasoar added the enhancement New/improved feature or request label Mar 6, 2026
@harveydevereux harveydevereux force-pushed the add_sevennet_to_train branch from 25efa36 to c917ba7 Compare March 12, 2026 11:35
@harveydevereux harveydevereux marked this pull request as ready for review March 12, 2026 11:36
harveydevereux and others added 2 commits March 13, 2026 08:06
Co-authored-by: Elliott Kasoar <45317199+ElliottKasoar@users.noreply.github.com>
oerc0122
oerc0122 previously approved these changes Mar 13, 2026
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.

Just some minor optional things, otherwise looks good.

Comment on lines +638 to +641
with open(metrics_path) as metrics:
lines = metrics.readlines()
assert len(lines) == 2
assert lines[0].split(",")[0] == "epoch"
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
with open(metrics_path) as metrics:
lines = metrics.readlines()
assert len(lines) == 2
assert lines[0].split(",")[0] == "epoch"
lines = metrics_path.read_text(encoding="utf-8")
assert len(lines) == 2
assert lines[0].split(",")[0] == "epoch"

Comment on lines +583 to +586
with open(metrics_path) as metrics:
lines = metrics.readlines()
assert len(lines) == 3
assert lines[0].split(",")[0] == "epoch"
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
with open(metrics_path) as metrics:
lines = metrics.readlines()
assert len(lines) == 3
assert lines[0].split(",")[0] == "epoch"
lines = metrics_path.read_text(encoding="utf-8")
assert len(lines) == 3
assert lines[0].split(",")[0] == "epoch"

Might as well do this

Co-authored-by: Jacob Wilkins <46597752+oerc0122@users.noreply.github.com>
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.

3 participants