Spaces:
Running
Running
Update my_model/config/fine_tuning_config.py
Browse files
my_model/config/fine_tuning_config.py
CHANGED
@@ -6,10 +6,11 @@ import os
|
|
6 |
# *** Dataset ***
|
7 |
# Base directory where the script is running
|
8 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
9 |
# Path to the folder containing the data files, relative to the configuration file
|
10 |
DATA_FOLDER = 'fine_tuning_data'
|
11 |
# Full path to the data folder
|
12 |
-
DATA_FOLDER_PATH = os.path.join(
|
13 |
# Path to the dataset file (CSV format)
|
14 |
DATASET_FILE = os.path.join(DATA_FOLDER_PATH, 'fine_tuning_data_yolov5.csv') # or 'fine_tuning_data_detic.csv'
|
15 |
|
|
|
6 |
# *** Dataset ***
|
7 |
# Base directory where the script is running
|
8 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
9 |
+
PARENT = os.path.dirname(BASE_DIR)
|
10 |
# Path to the folder containing the data files, relative to the configuration file
|
11 |
DATA_FOLDER = 'fine_tuning_data'
|
12 |
# Full path to the data folder
|
13 |
+
DATA_FOLDER_PATH = os.path.join(PARENT, 'fine_tuner', DATA_FOLDER)
|
14 |
# Path to the dataset file (CSV format)
|
15 |
DATASET_FILE = os.path.join(DATA_FOLDER_PATH, 'fine_tuning_data_yolov5.csv') # or 'fine_tuning_data_detic.csv'
|
16 |
|