Hello, I want to use your code for my experiment, but I cannot train ResNet and MobileNet with your max average approach.

I discovered that a model(ResNet or MobileNet) diverges when I train a model with Max Average Approach using your code.

I think that there are some experiment details what you did not mention in your paper such as batch size or warm-up in ResNet experiments.

I want to know specific experiment details. Can you tell me more specific experiments details?

Also, in MobileNet, there are 7 branches in your code although 3 branches in your paper.

I also want to use MobileNet v2 with 3 branches version code.
Can you upload your 3 branches version code if you allow to open to the public?

Thank you