minor

Author: vijaydwivedi75

.gitignore

@@ -9,30 +9,11 @@ gmail.py
 #SBM datasets
 data/SBMs/*.pkl
 
-#Superpixels datasets
-data/superpixels/*.pkl
-data/superpixels/*.zip
-PATH/
-
 #ZINC dataset
 data/molecules/*.pkl
 data/molecules/*.pickle
 data/molecules/*.zip
 data/molecules/zinc-full/*.pkl
 
-#TSP
-data/TSP/*.txt
-data/TSP/*.pkl
-data/TSP/*.zip
-data/TSP/pyconcorde/
-
-#COLLAB
-data/COLLAB/*
 dataset/
 
-#CSL
-data/CSL/*.pkl
-data/CSL/*.zip
-data/CSL/*.pt
-
-

train/train_molecules_graph_regression.py

@@ -8,7 +8,6 @@
 
 from train.metrics import MAE
 
-
 def train_epoch(model, optimizer, device, data_loader, epoch):
     model.train()
     epoch_loss = 0