bp

parallel_bp.py

@@ -68,7 +68,7 @@ class MeanField():
         for step in range(self.max_iter):
            tmp=cav_ij
            #print(tmp)
-           cav_ij = F.softmax(torch.spmm(spB1,torch.log(cav_ij @ self.W),dim=1)
+           cav_ij = F.softmax(torch.spmm(spB1,torch.log(cav_ij @ self.W)),dim=1)
            #print(cav_ij)
            differ=torch.max(torch.abs(cav_ij-tmp))
            #print(differ)