We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f694ac commit ea5908fCopy full SHA for ea5908f
src/weathergen/datasets/stream_data.py
@@ -127,7 +127,7 @@ def add_empty_target(self, fstep: int) -> None:
127
128
self.target_tokens[fstep] += [[torch.tensor([], dtype=torch.int32)]]
129
self.target_tokens_lens[fstep] += [torch.zeros([self.nhc_target], dtype=torch.int32)]
130
- self.target_coords[fstep] += [[torch.zeros((0, 106)) for _ in range(self.nhc_target)]]
+ self.target_coords[fstep] += [[torch.zeros((0, 105)) for _ in range(self.nhc_target)]]
131
self.target_coords_lens[fstep] += [torch.zeros([self.nhc_target], dtype=torch.int32)]
132
self.target_coords_raw[fstep] += [[torch.tensor([]) for _ in range(self.nhc_target)]]
133
self.target_times_raw[fstep] += [
0 commit comments