How can I get the weight of banzhafteacher?
darkpromise98 opened this issue · comments
Thanks for sharing a nice work! When I check the code, I find the config.estimator is None, which is the weight for self.banzhafteacher. So How can I get the checkpoint of banzhafteacher?
We are sorry that our GPUs are training other models, and I hope I can share this weight after the iclr deadline.
I'm sorry to reply to you so late. We release the code to train the estimator.
Line 20 in 1024c87
If you want to apply it to other tasks, you can refer to the code we provided to train on other datasets. We will also publish our trained weights soon.
We release our pre-trained estimator weights. If you want to apply a to other tasks, you can initialize a new estimator with the weights we provide. If you want better performance, you can train the estimator with a smaller learning rate and more epochs.