Fix to CallbackCost
Correctly use data_specs in default training algorithm