Skip to content

Commit 46ee287

Browse files
committed
revise graphmae attribute
1 parent 0e8c062 commit 46ee287

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

torch_molecule/encoder/graphmae/modeling_graphmae.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -75,16 +75,6 @@ class GraphMAEMolecularEncoder(BaseMolecularEncoder):
7575
model_name : str, default="GraphMAEMolecularEncoder"
7676
Name of the model.
7777
78-
Attributes
79-
----------
80-
fitting_loss : List[float]
81-
Training loss history.
82-
fitting_epoch : int
83-
Last epoch number from training.
84-
model_class : Type[GNN]
85-
Class of the underlying GNN model.
86-
87-
8878
Examples
8979
--------
9080
>>> from torch_molecule import GraphMAEMolecularEncoder

0 commit comments

Comments
 (0)