adding comparison for all learning rates TODO: if this is a good comparison for all learning rates, remove other datasets