请问这段代码是不是不兼容pytorch的dataset?想处理的数据集是dataset形式的,请问有什么办法吗?