From 4f4a3ea332895ab15ceb018b74ba5a8e42908b7d Mon Sep 17 00:00:00 2001 From: misko Date: Sat, 14 Sep 2024 15:47:55 +0000 Subject: [PATCH] fix beamnet checksum --- spf/notebooks/simple_train_filter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spf/notebooks/simple_train_filter.py b/spf/notebooks/simple_train_filter.py index 99dc5dd..655b762 100644 --- a/spf/notebooks/simple_train_filter.py +++ b/spf/notebooks/simple_train_filter.py @@ -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: