Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Rename the selection operator
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Use random.uniform discrete mode
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Pass kwargs directly
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Assure uniform choice of manitudes in TA
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
AutoAugment: coallsce pollicies
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Avoid unnecessary casts in color operations expressed in terms of arithm ops
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
AutoContrast, one less float32 intermediate result
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Expose select operator
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Review rework, usage lessons learnt from the expiriments
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Adjust the docs
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Better handling of non-unique names, better custom kwargs handling, improved error messaging
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Rework pretty_select and utils
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Linter complaints
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Private modules
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Simplify docs, simplify code
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Move policy to separate module
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
More (and more intelegible) docs
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Type annotations
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Improved error reporting for as_param
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
More specific callable annotations
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Polish names
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Better error messaging, polishing
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Better error-reporting, warnings, default random_sign
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Make Aug private
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Wrapper tests
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
More tests
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
More tests
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Test select
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
More tests
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Shape passing fix
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>