feat(pt): add dpa3 alpha descriptor #4476
Conversation
![github-advanced-security[bot]](https://avatars.githubusercontent.com/in/57789?s=60&v=4){kind=small}
|
|
||
| # deserialize repflow | ||
| statistic_repflows = repflow_variable.pop("@variables") | ||
| env_mat = repflow_variable.pop("env_mat") |
Check notice
Code scanning / CodeQL
Unused local variable Note
| # cast the input to internal precsion | ||
| extended_coord = extended_coord.to(dtype=self.prec) | ||
| nframes, nloc, nnei = nlist.shape | ||
| nall = extended_coord.view(nframes, -1).shape[1] // 3 |
Check notice
Code scanning / CodeQL
Unused local variable Note
| # h2: nb x nloc x nnei x 3 | ||
| # msk: nb x nloc x nnei | ||
| nb, nloc, nnei, _ = edge_ebd.shape | ||
| e_dim = edge_ebd.shape[-1] |
Check notice
Code scanning / CodeQL
Unused local variable Note
| def list_update_res_residual( | ||
| self, update_list: list[torch.Tensor], update_name: str = "node" | ||
| ) -> torch.Tensor: | ||
| nitem = len(update_list) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| [False], # use_econf_tebd | ||
| ): | ||
| dtype = PRECISION_DICT[prec] | ||
| rtol, atol = get_tols(prec) |
Check notice
Code scanning / CodeQL
Unused local variable Note test
| [False], # use_econf_tebd | ||
| ): | ||
| dtype = PRECISION_DICT[prec] | ||
| rtol, atol = get_tols(prec) |
Check notice
Code scanning / CodeQL
Unused local variable Note test
|
|
||
| dd0.repflows.mean = torch.tensor(davg, dtype=dtype, device=env.DEVICE) | ||
| dd0.repflows.stddev = torch.tensor(dstd, dtype=dtype, device=env.DEVICE) | ||
| model = torch.jit.script(dd0) |
Check notice
Code scanning / CodeQL
Unused local variable Note test
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## dpa3-alpha #4476 +/- ##
==============================================
+ Coverage 83.75% 84.46% +0.71%
==============================================
Files 667 674 +7
Lines 61514 62794 +1280
Branches 3486 3486
==============================================
+ Hits 51523 53042 +1519
+ Misses 8866 8625 -241
- Partials 1125 1127 +2 ☔ View full report in Codecov by Sentry. |
This PR is an early experimental preview version of DPA3. Significant changes may occur in subsequent updates. Please use with caution.