search

ArvinZhuang / DSI-transformers

A huggingface transformers implementation of "Transformer Memory as a Differentiable Search Index"
MIT License
155 stars 14 forks source link

when running to 1000 steps, error happen... #5

Open wxskl opened 1 year ago

wxskl commented 1 year ago

AttributeError: 'DataParallel' object has no attribute 'generate' image

ArvinZhuang commented 1 year ago

Hi, I'm not sure what is going on here, but you can safely ignore the eval during training by commenting out eval_dataset=eval_dataset, in the trainer argument for now.

yzc111 commented 8 months ago

Hello, this problem is solved?

yzc111 commented 8 months ago

you can use model.module.generate, this problem can be solved