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
Download Patch File Download Diff File Expand all files Collapse all files

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_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)