SPIN training test
输入“/”快速插入内容
SPIN training test
SPIN training test
用户4459用户44592024年5月28日修改
1.
Fe C， on A100（40G）， before optim (using torch.max)：
模型 | 框架 | 总时间（s/100steps） | data_time | forward_time | | | | optim_time | other_time (validation) | ||||
| | | | spin_preprocess_time | nlist_time | model_forward_time | spin_postprocess_time | | | ||||
dpa1(attn_layer=0) | pt | 2.20s (6.1h / 1m steps) | 0.07s | 0.02s | 0.47s | 0.83s | 0.02s | 0.67s | 0.36s | ||||
sea | pt | 7.15s (19.8h / 1m steps) | 0.11s | 0.01s | 0.44s | 3.13s(4种元素) | 0.02s | 3.28s | 0.59s | ||||
sea | tf | 1.50s (4.1h / 1m steps) | 0.01s | 1.36s | | | | | 0.12s | ||||
Fe C， on A100（40G）， after optim (using torch.amax)：
模型 | 框架 | 总时间（s/100steps） | data_time | forward_time | | | | optim_time | other_time (validation) | ||||
| | | | spin_preprocess_time | nlist_time | model_forward_time | spin_postprocess_time | | | ||||
dpa1(attn_layer=0) | pt | 1.38s (3.8h / 1m steps) | 0.05s | 0.02s | 0.14s | 0.53s | 0.02s | 0.51s | 0.14s | ||||
sea | pt | 6.97s (19.3h / 1m steps) | 0.11s | 0.01s | 0.22s | 3.07s | 0.02s | 3.20s | 0.59s | ||||
sea | tf | 1.50s (4.1h / 1m steps) | 0.01s | 1.36s | | | | | 0.12s | ||||
2.
Fe， on A100（40G），before optim (using torch.max)：
模型 | 框架 | 总时间（s/100steps） | data_time | forward_time | | | | optim_time | other_time (validation) | |||
| | | ||||||||||