diff --git a/data/dataset.py b/data/dataset.py index 5643cab..ef8a87f 100644 --- a/data/dataset.py +++ b/data/dataset.py @@ -141,7 +141,7 @@ class Dataset(Dataset): if self.cache is not None: if frame in self.cache.keys(): data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in self.cache[frame].items()} - data = { **item, **self.static_cache[frame.id] } + data = { **data, **self.static_cache[frame.id] } return data separator = '_' @@ -213,7 +213,7 @@ class Dataset(Dataset): self.cache[frame] = data data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in data.items()} - data = { **item, **self.static_cache[frame.id] } + data = { **data, **self.static_cache[frame.id] } return data