github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add support for threshold, ratio, and quantile on traditional LoRa

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-12 20:02:22 -06:00
parent 8d6edae9fd
commit a61914ddd9
2 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/ExtractLoconProcess.py
View File

@@ -35,7 +35,7 @@ class ExtractLoconProcess(BaseExtractProcess):
self.disable_cp = self.get_conf('disable_cp', False)
# set modes
if self.mode not in ['fixed', 'threshold', 'ratio', 'quantile']:
if self.mode not in list(mode_dict.keys()):
raise ValueError(f"Unknown mode: {self.mode}")
self.linear_param = self.get_conf('linear', mode_dict[self.mode]['linear'], as_type=mode_dict[self.mode]['type'])
self.conv_param = self.get_conf('conv', mode_dict[self.mode]['conv'], as_type=mode_dict[self.mode]['type'])