So… that ‘Different prediction results, update different weights in the model’ can be implemented only by setting SGD as the optimizer? Since predictions like [a, b, a, a, b] would update different part of model (part A or part B) due to the different elements of that.
Is my understanding right?