Too many compile error

[lxgyChen]

my pytorch version is 1.1.0

1. 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...)