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

3
cnn_training.py
View file

@ -75,6 +75,7 @@ def main(
pooling_type=str(config["pooling_type"]), pooling_type=str(config["pooling_type"]),
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"]),
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"]),
@ -118,6 +119,7 @@ def run_network(
pooling_type: str, pooling_type: str,
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,
scale_data: float, scale_data: float,
use_scheduler: bool, use_scheduler: bool,
use_adam: bool, use_adam: bool,
@ -246,6 +248,7 @@ def run_network(
conv_0_enable_softmax=conv_0_enable_softmax, conv_0_enable_softmax=conv_0_enable_softmax,
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,
).to(device) ).to(device)
logger.info(model) logger.info(model)

5
config.json
View file

@ -27,11 +27,12 @@
"scheduler_patience": 100, // (10) "scheduler_patience": 100, // (10)
"scheduler_threshold": 1e-5, // (1e-4) "scheduler_threshold": 1e-5, // (1e-4)
"minimum_learning_rate": 1e-10, "minimum_learning_rate": 1e-10,
"learning_rate": 1e-3, "learning_rate": 1e-4,
// <- LR Scheduler // <- LR Scheduler
"pooling_type": "max", // (max), average, none "pooling_type": "max", // (max), average, none
"conv_0_enable_softmax": true, // true, (false) "conv_0_enable_softmax": true, // true, (false)
"conv_0_power_softmax": 2.0, // 0.0 = Exp "conv_0_power_softmax": 0.0, // (0.0) = Exp
"conv_0_meanmode_softmax": false, // true, false
"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)