apply_gyro_dropout

composer.functional.apply_gyro_dropout(model, iters_per_epoch, max_epoch, p, sigma, tau)

Replaces all instances of torch.nn.Dropout with a GyroDropout.

By masking Dropout layer, this usually improves accuracy.