diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 31611a1ed..38f15fd3e 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -28,7 +28,7 @@ jobs: - name: Create Venv run: | pyenv local 3.8.11 - python3 -m venv venv --prompt ai8x-training + python -m venv venv --prompt ai8x-training - name: Activate Venv run: source venv/bin/activate - name: Install Dependencies @@ -62,7 +62,7 @@ jobs: - name: Create Venv run: | pyenv local 3.8.11 - python3 -m venv venv --prompt ai8x-training + python -m venv venv --prompt ai8x-training - name: Activate Venv run: source venv/bin/activate - name: Install Dependencies @@ -89,7 +89,7 @@ jobs: - name: Create Venv run: | pyenv local 3.8.11 - python3 -m venv venv --prompt ai8x-training + python -m venv venv --prompt ai8x-training - name: Activate Venv run: source venv/bin/activate - name: Install Dependencies @@ -125,7 +125,7 @@ jobs: - name: Create Venv run: | pyenv local 3.8.11 - python3 -m venv venv --prompt ai8x-training + python -m venv venv --prompt ai8x-training - name: Activate Venv run: source venv/bin/activate - name: Install Dependencies diff --git a/datasets/kws20.py b/datasets/kws20.py index 151853fd3..768e02103 100644 --- a/datasets/kws20.py +++ b/datasets/kws20.py @@ -166,7 +166,7 @@ def __parse_augmentation(self, augmentation): 'Using defaults: [Min:-0.1, Max: 0.1]') self.augmentation['shift'] = {'min': -0.1, 'max': 0.1} if 'strech' not in augmentation: - print('No key `strech` in input augmentation dictionary! ' + print('No key `stretch` in input augmentation dictionary! ' 'Using defaults: [Min: 0.8, Max: 1.3]') self.augmentation['strech'] = {'min': 0.8, 'max': 1.3} diff --git a/regression/create_onnx_script.py b/regression/create_onnx_script.py index d8564b2e1..5c64177c9 100644 --- a/regression/create_onnx_script.py +++ b/regression/create_onnx_script.py @@ -108,7 +108,6 @@ def time_stamp(): modelsearch = element[-4][3:] datasearch = element[-3].split('_')[0] if datasearch == dataset.split('_')[0] and modelsearch == model: - # model_paths.remove(tar) tar_path = tar timestamp = time_stamp() temp = ( diff --git a/regression/create_test_script.py b/regression/create_test_script.py index fb7362004..b8a174aaf 100644 --- a/regression/create_test_script.py +++ b/regression/create_test_script.py @@ -86,8 +86,9 @@ def joining(lst): try: temp[i+1] = str(config[log_data][log_model]["epoch"]) except KeyError: - # Handle the KeyError by assigning a dummy value - temp[i+1] = "10" + print(f"\033[93m\u26A0\033[0m Warning: {temp[j+1]} model is" + + " missing information in test configuration files.") + continue if '--deterministic' not in temp: temp.insert(-1, '--deterministic') @@ -98,8 +99,9 @@ def joining(lst): path_data = config[log_data]["datapath"] temp[i+1] = str(config[log_data][log_model]["epoch"]) except KeyError: - # Handle the KeyError by assigning a dummy value - path_data = "/data_ssd" + print(f"\033[93m\u26A0\033[0m Warning: {temp[j+1]} model is" + + " missing information in test configuration files.") + continue temp.insert(-1, '--data ' + path_data) temp.append("\n") diff --git a/regression/last_dev.py b/regression/last_dev.py index 3a1a9bb24..767eda8f2 100644 --- a/regression/last_dev.py +++ b/regression/last_dev.py @@ -94,8 +94,9 @@ def dev_scripts(script_pth, output_file_pth): try: temp[i+1] = str(config[log_data][log_model]["epoch"]) except KeyError: - # Handle the KeyError by assigning a dummy value - temp[i+1] = "10" + print(f"\033[93m\u26A0\033[0m Warning: {temp[j+1]} model is" + + " missing information in test configuration files.") + continue if '--deterministic' not in temp: temp.insert(-1, '--deterministic') @@ -106,8 +107,9 @@ def dev_scripts(script_pth, output_file_pth): path_data = config[log_data]["datapath"] temp[i+1] = str(config[log_data][log_model]["epoch"]) except KeyError: - # Handle the KeyError by assigning a dummy value - path_data = "/data_ssd" + print(f"\033[93m\u26A0\033[0m Warning: {temp[j+1]} model is" + + " missing information in test configuration files.") + continue temp.insert(-1, '--data ' + path_data) temp.append("\n") diff --git a/utils/object_detection_utils.py b/utils/object_detection_utils.py index ba75f8b22..ff2cc2a73 100644 --- a/utils/object_detection_utils.py +++ b/utils/object_detection_utils.py @@ -37,7 +37,7 @@ def collate_fn(batch): def check_target_exists(target_list): """ Checks whether any object exists in given target list - Object detection data laoders return target as + Object detection data loaders return target as target[0]: boxes list target[1]: labels list For images without any objects, these lists are both empty