Update alphafold3.py (#245)

amorehead · web-flow · commit 7783d3305618 · 2024-09-10T09:20:27.000-07:00
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -6749,7 +6749,6 @@ def forward(
 
         # otherwise, noise and make it learn to denoise
 
-        batch_size = atom_inputs.shape[0]
         calc_diffusion_loss = exists(atom_pos)
 
         if calc_diffusion_loss:
