Fixing Callback error when LRR is not used
Summary
This MR introduces the following changes
- Fixing an issue with the
MyCallback
function. If the LRR (learning rate reducer) is not used, thelogs
dict of the callback does not containlr
. Therefore the callback will crash with aKeyError
. This is fixed now.