diff --git a/README.md b/README.md index e3c3a8d..fa4c6a3 100644 --- a/README.md +++ b/README.md @@ -3,19 +3,20 @@ # Installation -```pip install k3-addons``` +```pip install k3-addons +``` # Includes: ## Layers -- Pooling: +- ### Pooling: - `k3_addons.layers.AdaptiveAveragePooling1D` - `k3_addons.layers.AdaptiveMaxPooling1D` - `k3_addons.layers.AdaptiveAveragePooling2D` - `k3_addons.layers.AdaptiveMaxPooling2D` - `k3_addons.layers.Maxout` -- Normalization +- #### Normalization - `k3_addons.layers.InstanceNormalization` -- Attention: +- #### Attention: - `k3_addons.layers.DoubleAttention` - `k3_addons.layers.AFTFull` - `k3_addons.layers.ChannelAttention2D` @@ -30,3 +31,18 @@ - `k3_addons.layers.MobileViTv2Attention` - `k3_addons.layers.ParNetAttention` - `k3_addons.layers.SimAM` + +## Losses + - `k3_addons.losses.ContrastiveLoss` + - `k3_addons.losses.GIoULoss` + - `k3_addons.losses.PinballLoss` + - `k3_addons.losses.SigmoidFocalCrossEntropy` + - `k3_addons.losses.WeightedKappaLoss` + - `k3_addons.losses.pairwise_distance` + - `k3_addons.losses.pinball_loss` +## Activations + - `k3_addons.activations.hardshrink` + - `k3_addons.activations.lisht` + - `k3_addons.activations.mish` + - `k3_addons.activations.snake` + - `k3_addons.activations.tanhshrink`