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 02:02:41 +02:00 committed by GitHub
parent 39b341f262
commit c849f2aca5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cnn_training.py
View file

@ -76,6 +76,7 @@ def main(
conv_0_enable_softmax=bool(config["conv_0_enable_softmax"]), conv_0_enable_softmax=bool(config["conv_0_enable_softmax"]),
conv_0_power_softmax=float(config["conv_0_power_softmax"]), conv_0_power_softmax=float(config["conv_0_power_softmax"]),
conv_0_meanmode_softmax=bool(config["conv_0_meanmode_softmax"]), conv_0_meanmode_softmax=bool(config["conv_0_meanmode_softmax"]),
conv_0_no_input_mode_softmax=bool(config["conv_0_no_input_mode_softmax"]),
scale_data=int(config["scale_data"]), scale_data=int(config["scale_data"]),
use_scheduler=bool(config["use_scheduler"]), use_scheduler=bool(config["use_scheduler"]),
use_adam=bool(config["use_adam"]), use_adam=bool(config["use_adam"]),
@ -120,6 +121,7 @@ def run_network(
conv_0_enable_softmax: bool, conv_0_enable_softmax: bool,
conv_0_power_softmax: float, conv_0_power_softmax: float,
conv_0_meanmode_softmax: bool, conv_0_meanmode_softmax: bool,
conv_0_no_input_mode_softmax: bool,
scale_data: float, scale_data: float,
use_scheduler: bool, use_scheduler: bool,
use_adam: bool, use_adam: bool,
@ -249,6 +251,7 @@ def run_network(
l_relu_negative_slope=leak_relu_negative_slope, l_relu_negative_slope=leak_relu_negative_slope,
conv_0_power_softmax=conv_0_power_softmax, conv_0_power_softmax=conv_0_power_softmax,
conv_0_meanmode_softmax=conv_0_meanmode_softmax, conv_0_meanmode_softmax=conv_0_meanmode_softmax,
conv_0_no_input_mode_softmax=conv_0_no_input_mode_softmax,
).to(device) ).to(device)
logger.info(model) logger.info(model)

5
config.json
View file

@ -29,10 +29,15 @@
"minimum_learning_rate": 1e-10, "minimum_learning_rate": 1e-10,
"learning_rate": 1e-4, "learning_rate": 1e-4,
// <- LR Scheduler // <- LR Scheduler
// type of pooling layer ->
"pooling_type": "max", // (max), average, none "pooling_type": "max", // (max), average, none
// <- type of pooling layer
// convolution layer 0 softmax parameters ->
"conv_0_enable_softmax": true, // true, (false) "conv_0_enable_softmax": true, // true, (false)
"conv_0_power_softmax": 0.0, // (0.0) = Exp "conv_0_power_softmax": 0.0, // (0.0) = Exp
"conv_0_meanmode_softmax": false, // true, false "conv_0_meanmode_softmax": false, // true, false
"conv_0_no_input_mode_softmax": false, // true, (false)
// <- convolution layer 0 softmax parameters
"use_adam": true, // (true) => adam, false => SGD "use_adam": true, // (true) => adam, false => SGD
"condition": "Coignless", "condition": "Coignless",
"scale_data": 255.0, // (255.0) "scale_data": 255.0, // (255.0)