Cope method not training

I have been trying to run cope method, both ways directly importing it from training, and using the plugin. But its not working well, there is issue is setting the p_size as it can only set as number of class in dataset, instead of last layer of model, Also there is problem is training of model, the method only runs on one sample of dataset, due to max_it_cnt being set 1 by default probably. It would be great to resolve it.