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 base_atomic_model.py

aa3365f
Select commit
Loading
Failed to load commit list.
Merged

pt: support multitask finetune #3480

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

94.36% of diff hit (target 77.48%)

View this Pull Request on Codecov

94.36% of diff hit (target 77.48%)

Annotations

Check warning on line 206 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#L206

Added line #L206 was not covered by tests

Check warning on line 210 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#L210

Added line #L210 was not covered by tests

Check warning on line 315 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#L315

Added line #L315 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 85 in deepmd/pt/utils/finetune.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L85

Added line #L85 was not covered by tests

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

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L138

Added line #L138 was not covered by tests

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

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/pt/utils/finetune.py#L156

Added line #L156 was not covered by tests