[Fix] save best checkpoint extra data

[joihn]

Motivation

Even when the user only save the best checkpoint, his preference about saving optimizer/scheduler should be be respected. Previously, scheduler/optimizer data were never saved.

Modification

Checkpoint saving logic