opened
juliusfrost
username_0
You can initialize a dataset pretty easily with `TrajectoryDataset()`. To add data either call `add()` or `append()`. Any data passed to those methods is downsized to save memory. To save the data to a directory call `save(save_location)`.  To load from a directory just call `load(save_location)`. You can also integrate the dataset with PyTorch's built-in DataLoader. You can check the test cases for examples.
Create Dataset class
issue