release the training code for private model?

[baogiadoan]

hi,

Can you also release the code to train the private model? You got better model performance compared to the previous target_model baselines.