Skip to content

Commit

Permalink
fix beamnet checksum
Browse files Browse the repository at this point in the history
  • Loading branch information
misko committed Sep 14, 2024
1 parent da5b899 commit 4f4a3ea
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion spf/notebooks/simple_train_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -922,7 +922,7 @@ def params_for_ds(ds, batch_size, resume_step):
)
]
if len(beamnet_params) > 0:
beamnet_checksum = torch.hstack().mean().item()
beamnet_checksum = torch.hstack(beamnet_params).mean().item()
print("Beamnet checksum:", beamnet_checksum)
if not math.isfinite(beamnet_checksum):
if args.wandb_project:
Expand Down

0 comments on commit 4f4a3ea

Please sign in to comment.