model= model.train(False)

    for  data in enumerate(test_loader):
       
        x, y_true = data 

        if torch.cuda.is_available():
            x, y_true = x.cuda(), y_true.cuda()

        out = model(x)
        out = F.softmax(out, dim=1)
        loss = criterion(out, y_true)

the error happens where i call the model (out=model(x)) in the testing section
the full error message is as follows:
TypeError: conv2d(): argument ‘input’ (position 1) must be Tensor, not int