Advanced Computing Platform for Theoretical Physics

Commit 2c10d31c authored by Pengfei Zhou's avatar Pengfei Zhou
Browse files

bp

parent a96d2da1
......@@ -74,7 +74,7 @@ class MeanField():
#print(differ)
if differ<self.conv_crit:
break;
marg_i = F.softmax(torch.spmm(spB1_,torch.log(tmp @ self.W))-torch.mean(marg_i@self.W,dim=0),dim=1)
marg_i = F.softmax(torch.spmm(spB1_,torch.log(cav_ij @ self.W))-torch.mean(marg_i@self.W,dim=0),dim=1)
print(step)
print(differ)
print(marg_i)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment