ITADN

修复一系列导致无法正常运行pre_train.py的BUG

#29Pull RequestLes1ie 创建于 2024-11-01已合并
external_contributor
L
Les1iecommented
修复一系列无法正常运行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 条评论