Change the display function to save instead of show

Add the training scripts for cloud training

Add the full training notebook