in dataset.py
Line 51: class TheDataset(data.dataset):
class TheDataset(data.dataset): TypeError: module() takes at most 2 arguments (3 given)
I solve it by changing to:class TheDataset(data.Dataset)
2.in train.py
Line 64: dataloader = torch.utils.data.dataLoader(train_set, batch_size=args.batch_size...)
dataloader = torch.utils.data.dataLoader(train_set, batch_size=args.batch_size, shuffle=True, AttributeError: module 'torch.utils.data' has no attribute 'dataLoader'
I solve it by changing to:dataloader = torch.utils.data.DataLoader(train_set, batch_size=args.batch_size...)
my pytorch version is 1.1.0
class TheDataset(data.dataset): TypeError: module() takes at most 2 arguments (3 given)I solve it by changing to:class TheDataset(data.Dataset)2.in train.py Line 64: dataloader = torch.utils.data.dataLoader(train_set, batch_size=args.batch_size...)
dataloader = torch.utils.data.dataLoader(train_set, batch_size=args.batch_size, shuffle=True, AttributeError: module 'torch.utils.data' has no attribute 'dataLoader'I solve it by changing to:dataloader = torch.utils.data.DataLoader(train_set, batch_size=args.batch_size...)