Merge pull request #3601 from shreyasvinaya/GANModel

GANModel Porting
deepchem · Nov 6, 2023 · 2c87061 · 2c87061
2 parents 657c3a7 + 1d8400e
commit 2c87061
Show file tree

Hide file tree

Showing 4 changed files with 595 additions and 18 deletions.
diff --git a/deepchem/models/torch_models/__init__.py b/deepchem/models/torch_models/__init__.py
@@ -9,7 +9,7 @@
 from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
 from deepchem.models.torch_models.gat import GAT, GATModel
 from deepchem.models.torch_models.gcn import GCN, GCNModel
-from deepchem.models.torch_models.gan import GAN
+from deepchem.models.torch_models.gan import GAN, GANModel
 from deepchem.models.torch_models.infograph import InfoGraphStar, InfoGraphStarModel, InfoGraphEncoder, GINEncoder, InfoGraph, InfoGraphModel, InfoGraphEncoder
 from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
 from deepchem.models.torch_models.lcnn import LCNN, LCNNModel