File size: 1,051 Bytes
24d7cbe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | # MatterSim 微调基础配置
# 请复制此文件后，修改下方带 "TODO:" 标记的路径为你本地实际路径。
# 数据路径（TODO: 修改为你的训练数据路径）
# 示例使用仓库自带的 data/high_level_water.xyz
train_data_path: ../data/high_level_water.xyz
# 验证数据路径，不需要可保持 null（TODO: 按需修改）
valid_data_path: null
# checkpoint 路径（TODO: 修改为你的预训练模型路径）
# 示例使用仓库自带的 weight/mattersim-v1.0.0-1M.pth
checkpoint: ../weight/mattersim-v1.0.0-1M.pth
# 输出设置
save_path: ./results/mattersim
run_name: onescience-mattersim
# 训练超参数
epochs: 1
batch_size: 16
lr: 2.0e-4
seed: 42
# 设备：cuda 或 cpu，留空则自动检测
device: cuda
# 训练目标
cutoff: 5.0
threebody_cutoff: 4.0
include_forces: true
include_stresses: false
# 损失权重
force_loss_ratio: 1.0
stress_loss_ratio: 0.1
# 保存与早停
save_checkpoint: true
ckpt_interval: 10
early_stop_patience: 10
# 重新归一化（可选）
re_normalize: false
|