我尝试从 torch.utils.data.dataloader
导入 default_collate
但它说 default_collate
不存在,但另一个函数确实存在存在(_collate_fn_t
)。
from torch.utils.data.dataloader import default_collate
from torch.utils.data.dataloader import _collate_fn_t
这些命令相似吗?
最佳答案
_collate_fn_t
是一般collate函数的类型定义,它定义为here如:
_collate_fn_t = Callable[[List[T]], Any]
default_collate
是 DataLoader 类使用的默认整理函数。
导入这些并不等同,您应该检查拼写并重试:
from torch.utils.data.dataloader import default_collate
关于python - 导入错误default_collate未定义,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62584755/