Hello,
l get the following error
    train(train_loader, model, criterion, optimizer, epoch)
TypeError: 'TensorDataset' object is not callable

when l run
    for epoch in range(args.start_epoch, args.epochs):
        if args.distributed:
            train_sampler.set_epoch(epoch)
        adjust_learning_rate(optimizer, epoch)

        # train for one epoch
        train(train_loader, model, criterion, optimizer, epoch)

the errors come from train_loader in train()
which is defined as follow :
 train_loader = torch.utils.data.DataLoader(
        train, batch_size=args.batch_size, shuffle=(train_sampler is None),
        num_workers=args.workers, pin_memory=True, sampler=train_sampler)

and the input   train    to train_loader is built as follow :
import torch.utils.data as data_utils
 data_train = torch.from_numpy(data_train)
labels_train=torch.from_numpy(labels_train)
train = data_utils.TensorDataset(data_train.float(), labels_train)

What’s wrong with my code ?
Thank you