Skip to content

Commit 3ede2af

Browse files
authored
Fix EpsilonPlusFlat and EpsilonAlpha2Beta1Flat composites (#147)
1 parent 12b3118 commit 3ede2af

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

src/lrp/composite_presets.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ function EpsilonPlusFlat(; epsilon=1.0f-6)
8484
ReshapingLayer => PassRule(),
8585
typeof(identity) => PassRule(),
8686
),
87-
FirstLayerTypeMap(ConvLayer => FlatRule(), Dense => FlatRule()),
87+
FirstLayerTypeMap(ConvLayer => FlatRule()),
8888
)
8989
end
9090

@@ -107,6 +107,6 @@ function EpsilonAlpha2Beta1Flat(; epsilon=1.0f-6)
107107
ReshapingLayer => PassRule(),
108108
typeof(identity) => PassRule(),
109109
),
110-
FirstLayerTypeMap(ConvLayer => FlatRule(), Dense => FlatRule()),
110+
FirstLayerTypeMap(ConvLayer => FlatRule()),
111111
)
112112
end

test/references/show/EpsilonAlpha2Beta1Flat.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,5 @@ Composite(
1616
Conv => FlatRule(),
1717
ConvTranspose => FlatRule(),
1818
CrossCor => FlatRule(),
19-
Dense => FlatRule(),
2019
),
2120
)

test/references/show/EpsilonPlusFlat.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,5 @@ Composite(
1616
Conv => FlatRule(),
1717
ConvTranspose => FlatRule(),
1818
CrossCor => FlatRule(),
19-
Dense => FlatRule(),
2019
),
2120
)

0 commit comments

Comments
 (0)