Solved by mahesh_bhosale in post #24 
              
              
                I have corrected the code to save dict of multiple LSTM Cells and load them individually like below, 

            # Save the checkpoint
            save_checkpoints({
                'num_epochs': epoch,
                'num_hidden': number_hidden,
                'num_cells': number_cells,
       …