Dataloader default collate_fn not working with square cutouts

[mb010]

The dataloader default collate_fn not working with square cutouts. It is unclear why this is happening. The error is:

 File "/opt/miniconda3/envs/astronavit/lib/python3.11/site-packages/torch/utils/data/_utils/collate.py", line 220, in collate_numpy_array_fn
    raise TypeError(default_collate_err_msg_format.format(elem.dtype))
TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found object

This might be solved by having target specific transforms which are applied to the target object, constraining it to be a torch tensor, but I didn't have time to check. The current plan is to write a collate_fn which is slightly more dynamic to catch this edge case.

[mb010]

#24 implemented a collate_fn for square images to temporarily address this issue.