seppl/optical-regeneration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor fixes and changes

Browse Source
This commit is contained in:
Joseph Hopfmüller
2024-11-29 15:49:10 +01:00
parent b156b9ceaf
commit ff32aefd52
3 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/single-core-regen/hypertraining/settings.py
View File

@@ -39,7 +39,7 @@ class PytorchSettings:
summary_dir: str = ".runs"
write_every: int = 10
head_symbols: int = 40
eye_symbols: int = 1000
eye_symbols: int = 400
# model settings
@@ -48,8 +48,11 @@ class ModelSettings:
output_dim: int = 2
n_hidden_layers: tuple | int = 3
n_hidden_nodes: tuple | int = 8
model_activation_func: tuple = "ModReLU"
model_activation_func: tuple | str = "ModReLU"
overrides: dict = field(default_factory=dict)
dropout_prob: float | None = None
model_layer_function: str | None = None
model_layer_parametrizations: list= field(default_factory=list)
@dataclass