diff --git a/helpers/app_settings.py b/helpers/app_settings.py index 382ffd4..b019f92 100644 --- a/helpers/app_settings.py +++ b/helpers/app_settings.py @@ -21,7 +21,7 @@ class DataSetConfig: beta: float sps: int snr_start: int - snr_end: int + snr_stop: int snr_step: int num_iterations: int recording_length: int diff --git a/requirements.txt b/requirements.txt index 191136c..9e5582d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,4 +10,4 @@ timm~=0.9 torch~=2.2 onnx~=1.18 onnxruntime~=1.22 -utils==0.1.2.dev0 +utils~=0.1.2.dev0