Would it better swap the `masked_tokens` and `unmasked_tokens`?

[He Cao]

Hi Phil:

I wonder whether here would it be better to swap the decoder_tokens and masked_tokens?

vit-pytorch/vit_pytorch/mae.py

Line 81 in 5ae5557

 decoder_tokens = torch.cat((decoder_tokens, mask_tokens), dim = 1) 

Since in Line55, the first part is masked_tokens and the second part is unmasked

vit-pytorch/vit_pytorch/mae.py

Line 55 in 5ae5557

 masked_indices, unmasked_indices = rand_indices[:, :num_masked], rand_indices[:, num_masked:] 

[Phil Wang]

@CiaoHe hello! 👋 sounds good, i made the change! 9f8c606