Skip to content

Commit

Permalink
fix(config): new results dir for issue #113
Browse files Browse the repository at this point in the history
  • Loading branch information
jyaacoub committed Jul 4, 2024
1 parent 099c3a3 commit b032b5b
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions src/utils/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,11 +87,16 @@ class LIG_FEAT_OPT(StringEnum):
DATA_ROOT = os.path.abspath('../data/')

# Model save paths
MEDIA_SAVE_DIR = os.path.abspath('results/model_media/')
MODEL_STATS_CSV = os.path.abspath('results/model_media/model_stats.csv')
MODEL_STATS_CSV_VAL = os.path.abspath('results/model_media/model_stats_val.csv')
MODEL_SAVE_DIR = os.path.abspath('results/model_checkpoints/ours')
issue_number = 113 # 113 is for unifying all splits for cross validation so that we are more confident
# when comparing results that they were trained in the same manner.
RESULTS_PATH = os.path.abspath(f'results/v{issue_number}/')
MEDIA_SAVE_DIR = f'{RESULTS_PATH}/model_media/'
MODEL_STATS_CSV = f'{RESULTS_PATH}/model_media/model_stats.csv'
MODEL_STATS_CSV_VAL = f'{RESULTS_PATH}/model_media/model_stats_val.csv'
MODEL_SAVE_DIR = f'{RESULTS_PATH}/model_checkpoints/ours'
CHECKPOINT_SAVE_DIR = MODEL_SAVE_DIR # alias for clarity
os.makedirs(MODEL_SAVE_DIR, exist_ok=True)
os.makedirs(MEDIA_SAVE_DIR, exist_ok=True)

# cluster based configs:
import socket
Expand Down

0 comments on commit b032b5b

Please sign in to comment.