Lightning-AI / Lightning-AI/pytorch-lightning

Automatic logger removal if the training failed (error happening on the sanity check)

Open
#15,655 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

feature logger won't fix
Dominant language
Python
Stars
31.4k
Forks
3.8k
Avg merge
6d 7h
Merged PRs (30d)
6

Description

## 🚀 Feature
Let lightning automatically remove Tensor board logger folder and files in case the training wasn't started successfully due to an error.

### Motivation
Right now lightning creates a logger file and then runs sanity checks ect.
If there are any bugs in the code, every run creates a new event file and a folder for it.
This leads to a stack of empty logger files that pile up and are cumbersome to remove manually, especially on a remote machine.

### Pitch
Logger folder gets removed if the error occurs before first successful training step in the epoch :)

cc @borda @awaelchli @Blaizzy

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

The issue names no files, tests, or entry points. Start by locating TensorBoard logger creation and the sanity-check/training lifecycle; done means a failed pre-training run removes its logger folder and files while a run that reaches a successful training step preserves them.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.