[Feature Request] _Replace with Title_

[zjp0803]

Summary

请问如何能从头训练unimol2

Detailed Description

我将--task 的参数改为unimol2，会在初始化LMDBDataset类时，在tmp_data_dir="/temp/“这个参数报错，看起来是有对train数据集进行分组处理，请问是需要自行分组还是您有已经处理好的数据集。我使用的是unimol仓库中贴出的molecular_property_prediction数据集。

Further Information, Files, and Links

No response