search

littlehacker26 / Discriminator-Cooperative-Unlikelihood-Prompt-Tuning

The code implementation of the EMNLP2022 paper: DisCup: Discriminator Cooperative Unlikelihood Prompt-tuning for Controllable Text Generation
25 stars 2 forks source link

torch.nn.modules.module.ModuleAttributeError: 'Prompt_tuning' object has no attribute 'predict' #3

Open mmmaxcc opened 9 months ago

mmmaxcc commented 9 months ago

when I try to run "bash train_sentiment_prompt.sh"，a error occured. Traceback (most recent call last): File "../main.py", line 161, in main() File "../main.py", line 150, in main trainer.train() File "/home/tjucic/bigdata/mxc/decup/Discriminator-Cooperative-Unlikelihood-Prompt-Tuning-master/trainer.py", line 192, in train result = self.evaluate(epoch_idx, 'Test') File "/home/tjucic/bigdata/mxc/decup/Discriminator-Cooperative-Unlikelihood-Prompt-Tuning-master/trainer.py", line 100, in evaluate pred_ids = self.model.predict(x,musk) File "/home/tjucic/anaconda3/envs/mxc/lib/python3.6/site-packages/torch/nn/modules/module.py", line 772, in getattr type(self).name, name)) torch.nn.modules.module.ModuleAttributeError: 'Prompt_tuning' object has no attribute 'predict'

mmmaxcc commented 9 months ago

ok I solved it。THX。

WaltzZY commented 8 months ago

ok I solved it。THX。

How did you solve it? Thank you so much