how to use test_data

[xiefei1026]

hi, I am a green hand in this project, emm, i just want to run a simple dataset to see the amaing effect runned by this project. should I replace the following code to read test_data? how to read it?

text = torch.randint(0, 49408, (4, 256)).cuda() images = torch.randn(4, 3, 256, 256).cuda()

Thanks for your help