Create checkpoints folders to save multiple steps checkpoints individually
Currently, we use a validation dataset on a graph built on train mode. However, this will not save the best performance model for the test dataset. During the training step, considering the memory issue, we cannot build up to two graphs for training and validation dataset simultaneously. The alternative way will save checkpoints on each certain steps, and then evaluate them on the postprocess step