Xiaoxia (Shirley) Wu
Xiaoxia (Shirley) Wu
tested the accuracy and looks great.
@ftian1, there is a formatting issue on the PR. The pre-commit needs to be run and the file changes committed to the branch. In particular, the following needs to be...
@JamesTheZ may know about this.
@madlag it looks like the prediction mapping is wrong? [0,1,2], ==>-1[2,0,1]: if you go to https://github.com/huggingface/nn_pruning/blob/main/examples/text_classification/glue_xp.py#L332 and and update the prediction ``` old_preds = preds.copy() preds = [] #[0,1,2] ==>-1[2,0,1]...