stochastic_depth.py 2 KB