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

set batch_y to 1 and want to test 15625

Browse Source
This commit is contained in:
mhz 2024-08-20 22:15:25 +02:00
parent 1fa2d49c11
commit 3950a8438d
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graph_dit/datasets/dataset.py
View File

@ -781,7 +781,7 @@ class Dataset(InMemoryDataset):
print(f'idx={idx}, y={y}') print(f'idx={idx}, y={y}')
y = torch.tensor([0, 0], dtype=torch.float).view(1, -1) y = torch.tensor([0, 0], dtype=torch.float).view(1, -1)
data = Data(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y, idx=i) data = Data(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y, idx=i)
return None # return None
return data return data
graph_list = [] graph_list = []
class Args: class Args:

3
graph_dit/diffusion_model.py
View File

@ -356,7 +356,8 @@ class Graph_DiT(pl.LightningModule):
to_generate = min(samples_left_to_generate, bs) to_generate = min(samples_left_to_generate, bs)
to_save = min(samples_left_to_save, bs) to_save = min(samples_left_to_save, bs)
chains_save = min(chains_left_to_save, bs) chains_save = min(chains_left_to_save, bs)
batch_y = test_y_collection[batch_id : batch_id + to_generate] # batch_y = test_y_collection[batch_id : batch_id + to_generate]
batch_y = torch.ones(to_generate, self.ydim_output, device=self.device)
cur_sample = self.sample_batch(batch_id, to_generate, batch_y, save_final=to_save, cur_sample = self.sample_batch(batch_id, to_generate, batch_y, save_final=to_save,
keep_chain=chains_save, number_chain_steps=self.number_chain_steps) keep_chain=chains_save, number_chain_steps=self.number_chain_steps)