解决问题
AttributeError: module 'torch.utils' has no attribute 'data'
解决思路
属性错误:模块的'torch.utils'没有属性'data'
解决方法
只需在顶行添加此代码即可！
import torch.utils.data # 新添加代码
将
import logging
import numpy as np
import torch
from . import ava_helper as ava_helper
from . import cv2_transform as cv2_transform
from . import transform as transform
from . import utils as utils
from .build import DATASET_REGISTRY
logger = logging.getLogger(__name__)
@DATASET_REGISTRY.register()
class Ava(torch.utils.data.Dataset):
"""
AVA Dataset
"""
def __init__(self, cfg, split):
self.cfg = cfg
self._split = split
更改为
import logging
import numpy as np
import torch
import torch.utils.data # 新添加代码
from . import ava_helper as ava_helper
from . import cv2_transform as cv2_transform
from . import transform as transform
from . import utils as utils
from .build import DATASET_REGISTRY
logger = logging.getLogger(__name__)
@DATASET_REGISTRY.register()
class Ava(torch.utils.data.Dataset):
"""
AVA Dataset
"""
def __init__(self, cfg, split):
self.cfg = cfg
self._split = split
