[BUG] OrderedQueue

[RandallBalestriero]

@huguesva : issue on save and load checkpoint dur to extra dim for the out tensor

RuntimeError: Error(s) in loading state_dict for Module:
        size mismatch for _callbacks_modules.ordered_queue_label.out: copying a param with shape torch.Size([20000]) from checkpoint, the shape in current model is torch.Size([20000, 1]).

[huguesva]

is it still an issue @RandallBalestriero ? if so I can have a look