diff --git a/README.md b/README.md index 9e3676d..e446cc0 100644 --- a/README.md +++ b/README.md @@ -6,17 +6,18 @@ `pip install k3-addons` # Includes: -- Layers - - Pooling: - - `AdaptiveAveragePooling1D` - - `AdaptiveMaxPooling1D` - - `AdaptiveAveragePooling2D` - - `AdaptiveMaxPooling2D` - - Attention: - - `DoubleAttention` - - `AFTFull` - - `ChannelAttention2D` - - `SpatialAttention2D` - - `ECAAttention` - - `ExternalAttention` - - `ResidualAttention` \ No newline at end of file +## Layers +- Pooling: + - `k3_addons.layers.AdaptiveAveragePooling1D` + - `k3_addons.layers.AdaptiveMaxPooling1D` + - `k3_addons.layers.AdaptiveAveragePooling2D` + - `k3_addons.layers.AdaptiveMaxPooling2D` +- Attention: + - `k3_addons.layers.DoubleAttention` + - `k3_addons.layers.AFTFull` + - `k3_addons.layers.ChannelAttention2D` + - `k3_addons.layers.SpatialAttention2D` + - `k3_addons.layers.ECAAttention` + - `k3_addons.layers.ExternalAttention` + - `k3_addons.layers.ResidualAttention` + - `k3_addons.layers.MobileViTAttention` \ No newline at end of file