UnitDropout

class core.transforms.UnitDropout(min_units)

Bases: object

A simple unit dropout transform.

It first samples the proportion of units to keep by uniformly sampling a float p in [min_units, 1.0]. Then it removes (1 - p) units randomly.

Parameters:

min_units (float) – Minimum proportion of units to keep