pycharm运行导入项目报错AttributeError: module 'torch.utils.data' has no attribute 'Dataset'

这是在GitHub上下载的一个项目，运行时一直报错

报错代码为：
Traceback (most recent call last):
File "C:\Users\XXX\PycharmProjects\pythonProject\venv\MeshCNN-master\train.py", line 3, in
from data import Dataloader
File "C:\Users\XX\PycharmProjects\pythonProject\venv\MeshCNN-master\data_init_.py", line 2, in
from data.base_dataset import collate_fn
File "C:\Users\XX\PycharmProjects\pythonProject\venv\MeshCNN-master\data\base_dataset.py", line 6, in
class BaseDataset(data.Dataset):
AttributeError: module 'torch.utils.data' has no attribute 'Dataset'

pytorch版本可能不同，导包有问题，导包那里加上这行试试
import torchvision.datasets as datasets