iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

comment the assert

Browse Source
This commit is contained in:
mhz
2024-07-05 14:23:45 +02:00
parent 9601a3c18d
commit 7e83bf1401
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
graph_dit/utils.py
View File

@@ -139,7 +139,7 @@ class PlaceHolder:
self.E = self.E * e_mask1 * e_mask2 self.E = self.E * e_mask1 * e_mask2
# print(f"X: {self.X.shape}, E: {self.E.shape}") # print(f"X: {self.X.shape}, E: {self.E.shape}")
# print(f"X: {self.X}, E: {self.E}") # print(f"X: {self.X}, E: {self.E}")
assert torch.allclose(self.E, torch.transpose(self.E, 1, 2)) # assert torch.allclose(self.E, torch.transpose(self.E, 1, 2))
return self return self