Can we change to use saveModule instead of save during checkpointing

[jenniew]

module.save use Java serialization and the saved model is incompatible in different BigDL version. If user upgrade their BigDL version, the previous checkpoint model cannot be used any more.