Skip to content

Commit 1d39571

Browse files
committed
Deleting res folder if already exists
1 parent 54a3888 commit 1d39571

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/thunder/benchmark.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,9 @@ def run_benchmark(cfg: DictConfig, model_cls: Callable = None) -> None:
170170
task_type,
171171
adaptation_type,
172172
)
173-
os.makedirs(res_folder, exist_ok=True)
173+
if os.path.exists(res_folder):
174+
shutil.rmtree(res_folder)
175+
os.makedirs(res_folder)
174176

175177
if task_type in ["linear_probing", "segmentation"]:
176178
# Model checkpoints folder

0 commit comments

Comments
 (0)