TypeError: TET_loss() missing 5 required positional arguments: 'outputs', 'labels', 'criterion', 'means', and 'lamb' criterion = TET_loss().cuda()

[OuYangLiang0509]

Hello author, very interested in your work, but I am tuning the imagenet code is the following error occurs, looking forward to your reply. Traceback (most recent call last): File "train_amp.py", line 212, in Traceback (most recent call last): File "train_amp.py", line 212, in criterion = TET_loss().cuda() TypeError: TET_loss() missing 5 required positional arguments: 'outputs', 'labels', 'criterion', 'means', and 'lamb' criterion = TET_loss().cuda() TypeError: TET_loss() missing 5 required positional arguments: 'outputs', 'labels', 'criterion', 'means', and 'lamb'