set batch_y to 1 and want to test 15625

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

View File

@ -781,7 +781,7 @@ class Dataset(InMemoryDataset):
print(f'idx={idx}, y={y}')
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)
return None
# return None
return data
graph_list = []
class Args:

View File

@ -356,7 +356,8 @@ class Graph_DiT(pl.LightningModule):
to_generate = min(samples_left_to_generate, bs)
to_save = min(samples_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,
keep_chain=chains_save, number_chain_steps=self.number_chain_steps)