From f958e2d4e5308b8796129e163685952942cf1b56 Mon Sep 17 00:00:00 2001 From: David Rotermund <54365609+davrot@users.noreply.github.com> Date: Sat, 30 Apr 2022 14:51:02 +0200 Subject: [PATCH] Update README.md --- README.md | 69 ++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 68 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 9363fbe..d7a37b1 100644 --- a/README.md +++ b/README.md @@ -27,5 +27,72 @@ pip install pybind11 The Makefile uses clang as a compiler. If you want something else then you need to change the Makefile. The SbS.py autodetectes if the required C++ .so modules are in the same directory as the SbS.py file. -# Parameters +# Parameters in JSON file +## network_structure (required!) +Parameters of the network. The details about its layers and the number of output neurons. + +number_of_output_neurons: int = field(default=0) +forward_neuron_numbers: list[list[int]] = field(default_factory=list) +is_pooling_layer: list[bool] = field(default_factory=list) + +forward_kernel_size: list[list[int]] = field(default_factory=list) +strides: list[list[int]] = field(default_factory=list) +dilation: list[list[int]] = field(default_factory=list) +padding: list[list[int]] = field(default_factory=list) + +w_trainable: list[bool] = field(default_factory=list) +eps_xy_trainable: list[bool] = field(default_factory=list) +eps_xy_mean: list[bool] = field(default_factory=list) + + +## learning_parameters +Parameter required for training + +learning_active: bool = field(default=True) + +loss_coeffs_mse: float = field(default=0.5) +loss_coeffs_kldiv: float = field(default=1.0) + +optimizer_name: str = field(default="Adam") +learning_rate_gamma_w: float = field(default=-1.0) +learning_rate_gamma_eps_xy: float = field(default=-1.0) +learning_rate_threshold_w: float = field(default=0.00001) +learning_rate_threshold_eps_xy: float = field(default=0.00001) + +lr_schedule_name: str = field(default="ReduceLROnPlateau") +lr_scheduler_factor_w: float = field(default=0.75) +lr_scheduler_patience_w: int = field(default=-1) + +lr_scheduler_factor_eps_xy: float = field(default=0.75) +lr_scheduler_patience_eps_xy: int = field(default=-1) + +number_of_batches_for_one_update: int = field(default=1) +overload_path: str = field(default="./Previous") + +weight_noise_amplitude: float = field(default=0.01) +eps_xy_intitial: float = field(default=0.1) + +test_every_x_learning_steps: int = field(default=50) +test_during_learning: bool = field(default=True) + +alpha_number_of_iterations: int = field(default=0) + +## augmentation +Parameters used for data augmentation. + +crop_width_in_pixel: int = field(default=2) + +flip_p: float = field(default=0.5) + +jitter_brightness: float = field(default=0.5) +jitter_contrast: float = field(default=0.1) +jitter_saturation: float = field(default=0.1) +jitter_hue: float = field(default=0.15) + + +## ImageStatistics (please ignore) +(Statistical) information about the input. i.e. mean values and the x and y size of the input + +mean: list[float] = field(default_factory=list) +the_size: list[int] = field(default_factory=list)