When training with 02_train_network.py, execution terminates with the error:
RuntimeError: Batches of sparse tensors are not currently supported by the default collate_fn; please provide a custom collate_fn to handle them appropriately.
I notice that train_loader is not passed collate_fn when batch_size==1.
This is on pytorch 2.4.0 and CUDA 12.4, which otherwise works up to that point.
When training with
02_train_network.py, execution terminates with the error:I notice that
train_loaderis not passedcollate_fnwhen batch_size==1.This is on pytorch 2.4.0 and CUDA 12.4, which otherwise works up to that point.