Nuscenes with Batch size > 1

[BarathwajAnandan]

Hi,
I managed to initiate training with Nuscenes but apparently if Batch size > 1, there's a 'NotImplementedError'

        if B == 1:
            new_reference_points = coords3d.unsqueeze(0)    # (B, M, 3)
        else:
            print("Batch B:", B)
            raise NotImplementedError

in models/de

by any chance do you have an update for this part or do you guys still train for Batch size = 1?

Thanks!