torch2.1中,torch._six已启用,那我需要用什么来替代dataset.py中的from torch._six import int_classes as _int_classes
时间: 2024-02-22 19:00:44 浏览: 153
利用torch.utils.data.Dataset自定义数据加载类
3星 · 编辑精心推荐
在torch2.1中,可以使用以下语句来替代dataset.py中的from torch._six import int_classes as _int_classes:
```python
from six import int_classes as _int_classes
```
这是因为在torch2.1中,torch._six模块已被移除,但可以使用six模块来代替。因此,可以使用上述语句导入int_classes。
阅读全文