update grid var to post subsampling

pdebench · Oct 3, 2024 · e58e698 · e58e698
1 parent b11eab3
commit e58e698
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pdebench/models/fno/utils.py b/pdebench/models/fno/utils.py
@@ -351,7 +351,7 @@ def __init__(self, filename,
                 gridx, gridy = np.array(f['0023']['grid']['x'], dtype=np.float32), np.array(f['0023']['grid']['y'], dtype=np.float32)
                 mgridX, mgridY = np.meshgrid(gridx, gridy, indexing='ij')
                 _grid = torch.stack((torch.from_numpy(mgridX), torch.from_numpy(mgridY)), axis=-1)
-                grid = _grid[::reduced_resolution, ::reduced_resolution, ...]
+                _grid = _grid[::reduced_resolution, ::reduced_resolution, ...]
                 _tsteps_t = torch.from_numpy(np.array(f['0023']['grid']['t'], dtype=np.float32))
                 tsteps_t = _tsteps_t[::reduced_resolution_t]
                 self.data = _data