Named Layers inside Generic Containers (#143)

LuxDL · Sep 1, 2022 · 3018707 · 3018707 · avik-pal · Sep 1, 2022
1 parent 94b7e99
commit 3018707
Show file tree

Hide file tree

Showing 9 changed files with 690 additions and 527 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,8 +1,13 @@
 # v0.4
 
+## v0.4.19
+
+  - Generic Container layers (like `Chain`, `Parallel`, etc.) can now used custom naming for
+    their internal layers.
+
 ## v0.4.17
 
-  - Major breakcing change in experimental Recurrent Cell Implementations.
+  - Major breaking change in experimental Recurrent Cell Implementations.
 
 ## v0.4.14
   - Deprecate `bias` in favor of `use_bias` for `RNNCell`.

diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Lux"
 uuid = "b2108857-7c20-44ae-9111-449ecde12c47"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "0.4.18"
+version = "0.4.19"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/Lux.jl b/src/Lux.jl
@@ -32,6 +32,7 @@ include("core.jl")
 include("adapt.jl")
 # Layer Implementations
 include("layers/basic.jl")
+include("layers/containers.jl")
 include("layers/normalize.jl")
 include("layers/conv.jl")
 include("layers/dropout.jl")