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

pt: support multitask finetune #3480

Merged
merged 17 commits into from
Mar 22, 2024

Update training.py

19b9e79
Select commit
Loading
Failed to load commit list.
Merged

pt: support multitask finetune #3480

Update training.py
19b9e79
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Mar 21, 2024 in 1s

91.81% of diff hit (target 77.48%)

View this Pull Request on Codecov

91.81% of diff hit (target 77.48%)

Annotations

Check warning on line 99 in deepmd/dpmodel/atomic_model/dp_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/dp_atomic_model.py#L99

Added line #L99 was not covered by tests

Check warning on line 105 in deepmd/dpmodel/atomic_model/dp_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/dp_atomic_model.py#L105

Added line #L105 was not covered by tests

Check warning on line 292 in deepmd/dpmodel/atomic_model/linear_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/linear_atomic_model.py#L291-L292

Added lines #L291 - L292 were not covered by tests

Check warning on line 297 in deepmd/dpmodel/atomic_model/linear_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/linear_atomic_model.py#L297

Added line #L297 was not covered by tests

Check warning on line 142 in deepmd/dpmodel/atomic_model/pairtab_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/pairtab_atomic_model.py#L142

Added line #L142 was not covered by tests

Check warning on line 146 in deepmd/dpmodel/atomic_model/pairtab_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/dpmodel/atomic_model/pairtab_atomic_model.py#L146

Added line #L146 was not covered by tests

Check warning on line 300 in deepmd/pt/model/atomic_model/base_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/model/atomic_model/base_atomic_model.py#L300

Added line #L300 was not covered by tests

Check warning on line 306 in deepmd/pt/model/atomic_model/linear_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/model/atomic_model/linear_atomic_model.py#L305-L306

Added lines #L305 - L306 were not covered by tests

Check warning on line 311 in deepmd/pt/model/atomic_model/linear_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/model/atomic_model/linear_atomic_model.py#L311

Added line #L311 was not covered by tests

Check warning on line 250 in deepmd/pt/model/atomic_model/pairtab_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/model/atomic_model/pairtab_atomic_model.py#L250

Added line #L250 was not covered by tests

Check warning on line 254 in deepmd/pt/model/atomic_model/pairtab_atomic_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/model/atomic_model/pairtab_atomic_model.py#L254

Added line #L254 was not covered by tests

Check warning on line 98 in deepmd/pt/utils/auto_batch_size.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/auto_batch_size.py#L98

Added line #L98 was not covered by tests

Check warning on line 87 in deepmd/pt/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L87

Added line #L87 was not covered by tests

Check warning on line 140 in deepmd/pt/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L140

Added line #L140 was not covered by tests

Check warning on line 158 in deepmd/pt/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L158

Added line #L158 was not covered by tests

Check warning on line 132 in deepmd/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/utils/finetune.py#L132

Added line #L132 was not covered by tests

Check warning on line 136 in deepmd/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/utils/finetune.py#L136

Added line #L136 was not covered by tests