IMG_20191118_203351__013784×1364 1020 KB

I want to train N copies of the model on different data. After that, I want to use their summed losses as the loss for the original model (parameter index 0 in the image) and update the original model’s parameters based on that summed loss.
Let me know if that cleared it up. I am implementing the adaption step from the MAML algorithm.
Thank you,