n_train: 4664
n_val: 393
n_test: 393
train_prop_model(learning_rate=0.0001,name="iComformer", dataset="megnet", prop=props[3], pyg_input=True, n_epochs=500, max_neighbors=25, cutoff=4.0, batch_size=64, use_lattice=True, output_dir="MP-OFM-shear-modulus", use_angle=True, save_dataloader=True)
Test MAE: 15.067037818166545
【无标题】eee
于 2024-10-05 09:24:42 首次发布