About the second stage is training ViT in an end-to-end manner.

[yhl2018]

About the second stage is training ViT in an end-to-end manner，What are the execution parameters？use the config file pretrain_r2r_e2e.json'，get an error： File "mtrand.pyx", line 915, in numpy.random.mtrand.RandomState.choice ValueError: 'a' cannot be empty unless no samples are taken