成功解决AttributeError: module 'torch.utils' has no attribute 'data'

1. import logging
2. import numpy as np
3. import torch
4. 
5. from . import ava_helper as ava_helper
6. from . import cv2_transform as cv2_transform
7. from . import transform as transform
8. from . import utils as utils
9. 
10. from .build import DATASET_REGISTRY
11. 
12. logger = logging.getLogger(__name__)
13. 
14. 
15. @DATASET_REGISTRY.register()
16. class Ava(torch.utils.data.Dataset):  
17. """
18.     AVA Dataset
19.     """
20. 
21. def __init__(self, cfg, split):
22.         self.cfg = cfg
23.         self._split = split

1. import logging
2. import numpy as np
3. import torch
4. import torch.utils.data     #  新添加代码
5. 
6. from . import ava_helper as ava_helper
7. from . import cv2_transform as cv2_transform
8. from . import transform as transform
9. from . import utils as utils
10. 
11. from .build import DATASET_REGISTRY
12. 
13. logger = logging.getLogger(__name__)
14. 
15. 
16. @DATASET_REGISTRY.register()
17. class Ava(torch.utils.data.Dataset):  
18. """
19.     AVA Dataset
20.     """
21. 
22. def __init__(self, cfg, split):
23.         self.cfg = cfg
24.         self._split = split