globus-labs / cascade

95%

LAST BUILD ON BRANCH main

branch: main

CHANGE BRANCH

x

Reset

• main
• academy-proto
• add-chgnet
• add-lda
• add-mace
• ani-ens
• ani-hpo
• ani-stress
• blending
• cascade-proto
• chgnet-proxima
• different-npt
• dynamics-interface
• ensemble-stress
• first-docs
• fix-cp2k-ci
• init-packmol
• limit-train-size
• mace-freeze
• mace-no-results
• mace-on-the-fly
• multi-formula-training
• multi-head
• new-calculator
• online-learning
• proto
• proxima-tests
• return-of-template
• round-0
• scale-ani
• serial-prototype
• serial-proxima
• stress
• to-nvt
• train-with-dbs
• use-pretrained-model
• use-torchani
• v0.0.1

Build # 14757320726

Build Type

push

github

Committed by web-flow

Commit Message

Multi-head replay finetuning (#67)

* Start interface for finetuning with replay

- Tool for pulling relevant frames out of initial dataset
- Start the datamodel for passing replay instructions

* Add routines for pulling replay data

* Bug fix: ensure entires are Python floats

* Introduce function for creating models with shared layers

* Introduce replay into training function

* Add cli args for freezing layers

* Add replay to the training CLI script

* Indentation issue: was adding another replay each epoch

* Use pytorch-ignite's logic for skipping epochs

* More identation issues.

What was I thinking!?

Run Details

81 of 85 new or added lines in 4 files covered. (95.29%)

982 of 1030 relevant lines covered (95.34%)

0.95 hits per line