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

5
config.json
View file

@ -27,11 +27,12 @@
"scheduler_patience": 100, // (10)
"scheduler_threshold": 1e-5, // (1e-4)
"minimum_learning_rate": 1e-10,
"learning_rate": 1e-3,
"learning_rate": 1e-4,
// <- LR Scheduler
"pooling_type": "max", // (max), average, none
"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
"condition": "Coignless",
"scale_data": 255.0, // (255.0)