Fix bugs

2021-04-26 21:44:03 +08:00
parent 8358d71cdf
commit d3371296a7
10 changed files with 270 additions and 264 deletions
--- a/lib/datasets/synthetic_env.py
+++ b/lib/datasets/synthetic_env.py
@@ -63,7 +63,7 @@ class SyntheticDEnv(data.Dataset):
        dataset = np.random.multivariate_normal(
            mean_list, cov_matrix, size=self._num_per_task
        )
-        return index, torch.Tensor(dataset)
+        return timestamp, torch.Tensor(dataset)

    def __len__(self):
        return len(self._timestamp_generator)