setting an array element with a sequence.

[BlankOtis]

how to solve this problem？
setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (6,) + inhomogeneous part.

[liansekongyk]

I also have this question
I change the file path and the train.py(RuntimeError: Expected a 'cuda' device type for generator but found 'cpu')

data_loader = data.DataLoader(dataset, args.batch_size,

#                               num_workers=args.num_workers,
#                               shuffle=True, collate_fn=detection_collate,   
#                               pin_memory=True)
data_loader = data.DataLoader(dataset, args.batch_size,
                              num_workers=args.num_workers,
                              shuffle=True, collate_fn=detection_collate,
                              pin_memory=True, generator=torch.Generator(device='cuda'))

I think there is no parameter changed

[liansekongyk]

what your cuda vision?

[sonukiller]

Downgrading numpy solved the issue for me
pip install numpy==1.21.6