add BGNN-Adv and BGNN-MLP methods on bipartite graph

Author: yifanfeng97

dhg/models/graphs/bgnn.py

@@ -70,6 +70,7 @@ def train_one_layer(
         lbl_real = torch.ones(X_true.shape[0]).to(device)
         lbl_fake = torch.zeros(X_true.shape[0]).to(device)
 
+        netG.train(), netD.train()
         for _ in range(max_epoch):
             X_real = X_true
             X_fake = mp_func(netG(X_other))
@@ -209,6 +210,7 @@ def train_one_layer(
 
         X_true, X_other = X_true.to(device), X_other.to(device)
 
+        netG.train(), netD.train()
         for _ in range(max_epoch):
             X_real = X_true
             X_fake = netD(mp_func(netG(X_other)))