Fix the bug on data provider checkpoint

For single worker case, check if the data_dict has the key
'label_batches'
For multi worker case, recover the full image batch information from
label batch list