fix: Revert "fix: mask-out test labels during gradient updates in training loop (#728)" which has negative impact for roc/auc (#764)

This reverts commit 95f650d.

Author: zxkane

src/sagemaker/FD_SL_DGL/gnn_fraud_detection_dgl/fd_sl_train_entry_point.py

@@ -35,20 +35,14 @@ def train_fg(model, optim, loss, features, labels, train_g, test_g, test_mask,
     A full graph verison of RGCN training
     """
 
-    # get list of indecies for train labels
-    train_mask= th.logical_not(test_mask)
-    train_idx=  th.nonzero(train_mask, as_tuple=True)[0]
-    
     duration = []
     for epoch in range(n_epochs):
         tic = time.time()
         loss_val = 0.
 
         pred = model(train_g, features.to(device))
 
-        # only compute gradient updates for labels in train split
-        l = loss(th.index_select(pred, 0, train_idx), 
-                 th.index_select(labels, 0, train_idx))
+        l = loss(pred, labels)
 
         optim.zero_grad()
         l.backward()
@@ -262,4 +256,4 @@ def get_model(ntype_dict, etypes, hyperparams, in_feats, n_classes, device):
 
     print("Saving model")
     save_model(g, model, args.model_dir, id_to_node, mean, stdev)
-    print("Model and metadata saved")
+    print("Model and metadata saved")