Commit e48497c
Han Wang
fix: add missing accessor methods to DescrptBlockRepformers
The env_mat_stat code calls descriptor.get_rcut_smth() and
descriptor.get_env_protection() during compute_input_stats, but
DescrptBlockRepformers only had the attributes without accessor
methods, causing AttributeError when training DPA2 models.1 parent 4ebce58 commit e48497c
1 file changed
+8
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
345 | 345 | | |
346 | 346 | | |
347 | 347 | | |
| 348 | + | |
| 349 | + | |
| 350 | + | |
| 351 | + | |
| 352 | + | |
| 353 | + | |
| 354 | + | |
| 355 | + | |
348 | 356 | | |
349 | 357 | | |
350 | 358 | | |
| |||
0 commit comments