xarray-contrib/xbatcher

Support xarray.DataArray and xarray.Dataset batches in PyTorch dataloader

maxrjones opened this issue · 2 comments

maxrjones commented

As discussed in #71 (comment)_, the PyTorch DataLoader should be extended to support either dataarray or dataset batches after the batch generator output was generalized in #71.

weiji14 commented

I can work on this, got some code already to improve the .torch accessor, can you assign me this issue?

maxrjones commented

I can work on this, got some code already to improve the .torch accessor, can you assign me this issue?

Super, thanks! Assigned