-
Notifications
You must be signed in to change notification settings - Fork 593
Closed
Description
not sure how to handle type embedding - type embedding is not a model parameter,
but instead a part of the input data. Maybe the interface should be refactored...
Line: 1429
deepmd-kit//deepmd/tf/descriptor/se_a.py
Lines 1426 to 1436 in 76371ce
| raise NotImplementedError("spin is unsupported") | |
| assert self.davg is not None | |
| assert self.dstd is not None | |
| # TODO: tf: handle type embedding in DescrptSeA.serialize | |
| # not sure how to handle type embedding - type embedding is not a model parameter, | |
| # but instead a part of the input data. Maybe the interface should be refactored... | |
| return { | |
| "@class": "Descriptor", | |
| "type": "se_e2_a", | |
| "@version": 1, |
Reactions are currently unavailable