search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.21k stars 1.27k forks source link

fine-tune XLNet model #513

Open alshahrani2030 opened 4 years ago

alshahrani2030 commented 4 years ago

I am trying to fine-tune XLNet model and it used work fine but I think huggingface update some classes and I ran through this error:

RuntimeError: Trying to create tensor with negative dimension -1: [-1, 768] xlnet

zylo117 commented 4 years ago

?