[Contributions Welcome] TQDM Progbar doesn't reset after KeyboardInterrupt

[ncullen93]

[Contributions Welcome] TQDM Progbar doesn't reset after KeyboardInterrupt

[ncullen93]

fixed thanks @recastrodiaz

[recastrodiaz]

I'm not sure why this happens but I'd love to get this fixed too. In my case, I don't need to KeyboardInterrupt the training. Just letting it complete and running it again seems to cause the issue (not always). In some cases, if I run the previous 2 cells, the loader loads fine. But once it happens, only a kernel restart seems to fix it.

This issue is on TQDM's tracker too:

• tqdm/tqdm#375
• tqdm/tqdm#358

[recastrodiaz]

This race condition might be related: tqdm/tqdm#348

[recastrodiaz]

I've figured out the second issue. Running fit_loader several times adds multiple instances of TQDM to self._callbacks, printing N times over the same row, corrupting the progress bar:

https://github.com/ncullen93/torchsample/blob/2154d789ef0d79debe2cc522751ae7072e8b22dc/torchsample/modules/super_module.py#L155

The code should be instead:

        progressbar = []
        if verbose > 0:
            progressbar += [TQDM()]
        callbacks = CallbackModule(self._callbacks + progressbar)

From:

        if verbose > 0:
            self._callbacks += [TQDM()]
        callbacks = CallbackModule(self._callbacks)

As for the first issue (KeyboardInterrupt) a with statement should suffice. Trying that out as I type this.