scheduler_type (str, optional) —
The type of scheduler to load. If not provided, the scheduler type will be inferred from the checkpoint file.   prediction_type (str, optional) —
The type of prediction to load. If not provided, the prediction type will be inferred from the checkpoint file.   kwargs (remaining dictionary of keyword arguments, optional) —
Can be used to overwrite load and saveable variables (the pipeline components of the specifi