修复一系列导致无法正常运行pre_train.py的BUG
external_contributor
修复一系列无法正常运行pre_train.py的BUG:
- 修复pre_train.py中,pretrain_task参数名不匹配的错误
- 修复SimGRACE在初始化NodePretrain时dataname参数的错误
- 在PreTrain中接受num_workers参数,并在子类初始化Dataloader时使用
- 修复GraphPrePrompt、NodePrePrompt无法正确识别device的错误
合并状态:已合并 合并于 2024-11-02 关闭于 2024-11-02 0 条评论