GoogLeNet attaches two auxiliary classifiers to middle layers and adds their losses to the main loss with weight 0.3. The extra signal helps gradients reach the early layers of a very deep network.
Implement weighted_total_loss(main, aux, w), returning the main loss plus the weighted auxiliary loss.
Multiply the auxiliary loss by w and add it to the main loss.
No results yet
Run your code to check it against the examples, or submit to run every test.