question about training

[jzssz]

Dear Bao: When I have trained my u-vit model via train_t2i_discrete.py in U-Vit code, I get 3 model files in folder '/ckpts/' including 'nnet.pth' 'nnet_ema.pth' 'optimizer.pth', which file is used to integrate in unidiffuser as the role 'uvit_v0/1.pth' plays?

Much appreciate if you can take a look sometime!