Training in a custom dataset

[Zumbalamambo]

How do I train it in a custom dataset?

[ByungSun12]

Unfortunately, You need to implement multi-staged backpropagation in your self. Authors said that it is easy to implement multi-staged backpropagation. You can see pseudo code in CVPR's supplementary.

The supplementary is in the conference proceedings: http://openaccess.thecvf.com/content_ICCV_2019/supplemental/Revaud_Learning_With_Average_ICCV_2019_supplemental.pdf