When I try to train with lsp,I faced some problem

[GuoHaiYang123]

TypeError: can't assign a tuple to a torch.FloatTensor

[Github3G]

TypeError: can't assign a tuple to a torch.FloatTensor

I meet the same question. and i find there is not a 'target_weight' in the trainloader at the 194 line of main.py.
And have you resolved this problem?

[Polaris7777777]

Have you addressed the problem with target_weight? If you have, could you please provide me with some instructions? Thank you very much.