Allow training_step
in manual optimization to return general mappin…
#3172
Loading
training_step
in manual optimization to return general mappin…
#3172