Size Mismatch

[SoumyaDC02012309]

raise ValueError(f"Target size ({target.size()}) must be the same as input size ({input.size()})")
ValueError: Target size (torch.Size([128, 300, 10])) must be the same as input size (torch.Size([128, 300, 4]))

How to take care of this while using USPTO50K

Thanks in advance

[lanna0504]

raise ValueError(f"Target size ({target.size()}) must be the same as input size ({input.size()})") ValueError: Target size (torch.Size([128, 300, 10])) must be the same as input size (torch.Size([128, 300, 4]))

How to take care of this while using USPTO50K

Thanks in advance

It seems that this issue has been reported by others before, and the solution is to set the default of "max_ct_atom" to 10 instead of 4. I did this and could train the model successfully.