Multiple Demos Loading

[SantiDiazC]

Hi, I found your code interesting. I am working on an imitation learning task for controlling a robot arm which requires multiple demonstrations. As you did in your project you load the video records and car commands store in the hd5f provided in the carla dataset, however I was not able to follow your code on how to load each different demonstration at each episode. Can you provide me a simple explanation of that in your code or can you help me with some documentation on that please? I can't find yet a good documentation on how to do that with pytorch. Thank you in advance!