forward_features(self, x, keep_rate=None, tokens=None, get_idx=False)

[60wanjinbing]

with torch.cuda.amp.autocast(): outputs = model(samples, keep_rate) loss = criterion(samples, outputs, targets) 这段代码是在train_one_epoch函数中调用的，你的model没有传token参数，按照你这个应该会报错的，请问你这个token是在哪里传过去的？