Thomas Kelly
Thomas Kelly
Using generate returns the auto-regressive attentions. Try using teacher forcing (ie providing labels/decoder_ids) through the forward pass
This is a common and well-known error for the Delaney dataset.. Most of the time it is solved by ensuring at train time, by ensuring that both input and labels...
Seems related #17043
Similar issue [#42 ](https://github.com/gcorso/DiffDock/issues/42#issuecomment-1312430151)
I think the error comes from a cropping function https://github.com/gcorso/DiffDock/blob/6d065cd78c11a610a0f3eba30f836a7e3a04358f/utils/sampling.py#L108