模型训练过程汇总[该仓库只含有graph model的训练过程]
本仓库采用扁平化的目录结构和标签系统来组织模型,具体说明如下:
一级目录:直接以模型名称-数据集,例如 ResNet-CIFAR-10、GraphMAE_QM9-Cora 等
二级目录:包含该模型在该数据集下的不同训练任务或变体,例如 normal、noisy、backdoor_invisible 等
训练过程目录结构:每个模型目录下包含:
scripts/:存放模型相关代码和训练脚本
epochs/:存放模型训练过程和权重文件
每个epoch的权重文件(model.pth)和embedding(.npy)
dataset/:模型需要的数据集