clean

vijaydwivedi75 · vijaydwivedi75 · commit 6d3a48e6d318 · 2020-12-18T01:05:44.000+08:00
diff --git a/data/molecules.py b/data/molecules.py
@@ -239,7 +239,7 @@ class MoleculeDataset(torch.utils.data.Dataset):
 
     def __init__(self, name):
         """
-            Loading SBM datasets
+            Loading ZINC dataset
         """
         start = time.time()
         print("[I] Loading dataset %s..." % (name))
diff --git a/environment_cpu.yml b/environment_cpu.yml
@@ -40,5 +40,4 @@ dependencies:
 - pip:
   - tensorflow==2.1.0
   - tensorflow-estimator==2.1.0
-  - tensorboard==2.1.1
-  - ogb==1.2.3
+  - tensorboard==2.1.1
diff --git a/environment_gpu.yml b/environment_gpu.yml
@@ -43,5 +43,4 @@ dependencies:
 - pip:
   - tensorflow-gpu==2.1.0
   - tensorflow-estimator==2.1.0
-  - tensorboard==2.1.1
-  - ogb==1.2.3
+  - tensorboard==2.1.1
