TextBox train.py error and possible solutions
hanchi-gao opened this issue · 1 comments
get the error In follow image
File "/textbox/Textbox_test/TextBox/textbox/trainer/trainer.py", line 516, in evaluate generated = self.accelerator.unwrap_model(self.model).generate(batch_data, self.accelerator) TypeError: generate() missing 1 required positional argument: 'accelerator'
I compared the older version:
latest version
line 516: generated = self.accelerator.unwrap_model(self.model).generate(batch_data, self.accelerator
older version
line 505: generated = self.accelerator.unwrap_model(self.model).generate(batch_data, eval_data, self.accelerator)
so, I added the argument "eval_data" to generate().
Sure enough, problem solved.
Please confirm this error
Thank you for pointing out this issue. We have removed the argument eval_data
in the generate
function.