Skip to content

Commit ae90498

Browse files
wanghan-iapcmHan Wang
andauthored
refactorize networks, now can be used cross platform (#3141)
Co-authored-by: Han Wang <[email protected]>
1 parent 15117a0 commit ae90498

File tree

2 files changed

+248
-227
lines changed

2 files changed

+248
-227
lines changed

deepmd_utils/model_format/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@
1414
NativeNet,
1515
NetworkCollection,
1616
load_dp_model,
17+
make_embedding_network,
18+
make_fitting_network,
19+
make_multilayer_network,
1720
save_dp_model,
1821
traverse_model_dict,
1922
)
@@ -32,6 +35,9 @@
3235
__all__ = [
3336
"DescrptSeA",
3437
"EnvMat",
38+
"make_multilayer_network",
39+
"make_embedding_network",
40+
"make_fitting_network",
3541
"EmbeddingNet",
3642
"FittingNet",
3743
"NativeLayer",

0 commit comments

Comments
 (0)