Running the example you gave in colab will give the following error, is your model inconsistent with the configuration file you gave?
You passed along num_labels=1200 with an incompatible id to label map: {'0': 'LABEL_0', '1': 'LABEL_1'}. The number of labels wil be overwritten to 2.
ValueError Traceback (most recent call last)
in <cell line: 0>()
1 from deepstyle.model import DeepStyle
2 # Give the folder of the model:
----> 3 m = DeepStyle("/content/drive/MyDrive/Colab Notebooks/写作风格嵌入/deepstyle")
4 # Sample document:
5 doc = "Welcome to day two of cold, nasty, wet weather. Ick. Rain is so bad by itself... But when you mix it with a hella cold temperature and nasty wind... Not so much fun anymore."
7 frames
/usr/local/lib/python3.11/dist-packages/numpy/core/fromnumeric.py in _wrapfunc(obj, method, *args, **kwds)
57
58 try:
---> 59 return bound(*args, **kwds)
60 except TypeError:
61 # A TypeError occurs if the object does have such a method in its
ValueError: cannot reshape array of size 921600 into shape (768,2)