Recommend change the code

[zlwangustc]

recommend change the following code int lib/models/cls_cvt.py:611
x = torch.squeeze(x) change to x = torch.squeeze(x, dim=1)
Otherwise, an error will occur when bachsize = 1