AttributeError: ‘VotingClassifier‘ object has no attribute ‘set_criterion‘

今天在跑torchensemble的example的石斛出现了下面的问题：

p-2021_11_13_21_52.log...
Traceback (most recent call last):
  File "voting_demo.py", line 48, in <module>
    model.set_criterion(criterion)
  File "/home/eric/anaconda3/envs/seg/lib/python3.6/site-packages/torch/nn/modules/module.py", line 594, in __getattr__
    type(self).__name__, name))
AttributeError: 'VotingClassifier' object has no attribute 'set_criterion'

解决方法

pip uninstall torchensemble 
 git clone https://github.com.cnpmjs.org/TorchEnsemble-Community/Ensemble-Pytorch.git
 cd Ensemble-Pytorch
 python setup.py install