davrot/kk_contour_net_shallow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-07-23 01:59:13 +02:00 committed by GitHub
parent 2bcb14def3
commit 39b341f262
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
functions/make_cnn.py
View file

@ -17,6 +17,7 @@ def make_cnn(
conv_0_enable_softmax: bool,
conv_0_power_softmax: float,
conv_0_meanmode_softmax: bool,
conv_0_no_input_mode_softmax: bool,
l_relu_negative_slope: float,
) -> torch.nn.Sequential:
assert len(conv_out_channels_list) >= 1
@ -63,7 +64,10 @@ def make_cnn(
if conv_0_enable_softmax:
cnn.append(
SoftmaxPower(
dim=1, power=conv_0_power_softmax, mean_mode=conv_0_meanmode_softmax
dim=1,
power=conv_0_power_softmax,
mean_mode=conv_0_meanmode_softmax,
no_input_mode=conv_0_no_input_mode_softmax,
)
)