Do not pass dataset metadata to model

LTMeyer · LTMeyer · commit ad1e21a86b5c · 2025-03-25T11:51:32.000+01:00
diff --git a/the_well/benchmark/train.py b/the_well/benchmark/train.py
@@ -57,7 +57,8 @@ def train(
     )
     model: torch.nn.Module = instantiate(
         cfg.model,
-        dset_metadata=dset_metadata,
+        n_spatial_dims=dset_metadata.n_spatial_dims,
+        spatial_resolution=dset_metadata.spatial_resolution,
         dim_in=n_input_fields,
         dim_out=n_output_fields,
     )

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,8 @@ def train(`
`57`	`57`	`)`
`58`	`58`	`model: torch.nn.Module = instantiate(`
`59`	`59`	`cfg.model,`
`60`		`- dset_metadata=dset_metadata,`
	`60`	`+ n_spatial_dims=dset_metadata.n_spatial_dims,`
	`61`	`+ spatial_resolution=dset_metadata.spatial_resolution,`
`61`	`62`	`dim_in=n_input_fields,`
`62`	`63`	`dim_out=n_output_fields,`
`63`	`64`	`)`