Update train.py

train.py

@@ -37,7 +37,7 @@ def train(args):
     test_data = Dataset(root=args.train_data_path, transform=preprocess, target_transform=target_transform, dataset_name = args.dataset)
     train_data = Dataset(root=args.train_data_path, transform=preprocess, target_transform=target_transform, dataset_name = args.dataset, mode='train')
     train_data = torch.utils.data.ConcatDataset([train_data, test_data])
-    print('Dataset length:', train_data.length)
+    print('Dataset length:', len(train_data))
     train_dataloader = torch.utils.data.DataLoader(train_data, batch_size=args.batch_size, shuffle=True)
 
   ##########################################################################################