diff --git a/src/dopt_sensor_anomalies/constants.py b/src/dopt_sensor_anomalies/constants.py index 43fcef7..a8493da 100644 --- a/src/dopt_sensor_anomalies/constants.py +++ b/src/dopt_sensor_anomalies/constants.py @@ -6,9 +6,9 @@ STOP_FOLDER_NAME: Final[str] = "python" MODEL_FOLDER_NAME: Final[str] = "models" THRESHOLD_BW: Final[int] = 63 -BACKBONE: Final[str] = "resnet18" -LAYERS: Final[tuple[str, str]] = ("layer1", "layer2") -RATIO: Final[float] = 0.05 +BACKBONE: Final[str] = "'wide_resnet50_2" +LAYERS: Final[tuple[str, str]] = ("layer1", "layer2", "layer3") +RATIO: Final[float] = 0.01 NUM_VALID_ELECTRODES: Final[int] = 6 HEATMAP_FILENAME_SUFFIX: Final[str] = "_Heatmap"