train_prior.py 5.51 KB