add checkpoints for plotting

bccc6c4

mingyuanzhou commited on Oct 27, 2023