Error at JointDataset?

[AI-Hunter]

@CaptainEven
Thanks for your nice repo.
There are two dataset preparation/loader:
MultiScaleJD ans JointDataset

When I am using JointDataset, it produce the following error:

    scale_idx = data_loader.dataset.batch_i_to_scale_i[batch_i]
AttributeError: 'JointDataset' object has no attribute 'batch_i_to_scale_i'

Would you give an idea, how can I solve it?
When can I use JointDataset and MultiScaleJD?
Thanks

[wsw321]

have you solve this problem? I meet this problem now.