depth is constant each width

[yumion]

Hello, thank you for releasing a implementation.

In my understanding, we change number of op as depth each width , however this code should be same depth each width here.

augmix/augment_and_mix.py

Line 61 in 7c84885

depth = depth if depth > 0 else np.random.randint(1, 4)

So, we change code like below, right?

depth = np.random.randint(1, depth + 1 if depth > 0 else 4)

Best, regard.

[normster]

Thanks for pointing this out, you're completely right. Feel free to submit a PR or I can make the change when I get a chance.

[normster]

Hi @yumion, I took a closer look and the suggested fix doesn't address the bug of overwriting depth on the first branch.