m7mdal7aj commited on
Commit
a1f1f0b
1 Parent(s): 6e570db

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(BASE_DIR, DATA_FOLDER)
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