|
|
" Utils for modifying what is displayed in notebooks and command line" |
|
|
import fastai |
|
|
import fastprogress |
|
|
|
|
|
from ..basic_train import * |
|
|
from ..core import * |
|
|
|
|
|
__all__ = ['progress_disabled_ctx'] |
|
|
|
|
|
class progress_disabled_ctx(): |
|
|
"Context manager to disable the progress update bar and Recorder print." |
|
|
def __init__(self,learn:Learner): |
|
|
self.learn = learn |
|
|
|
|
|
def __enter__(self): |
|
|
|
|
|
fastprogress.fastprogress.NO_BAR = True |
|
|
fastai.basic_train.master_bar,fastai.basic_train.progress_bar = fastprogress.force_console_behavior() |
|
|
self.orig_callback_fns = copy(self.learn.callback_fns) |
|
|
rec_name = [x for x in self.learn.callback_fns if hasattr(x, 'func') and x.func == Recorder] |
|
|
if len(rec_name): |
|
|
rec_idx = self.learn.callback_fns.index(rec_name[0]) |
|
|
self.learn.callback_fns[rec_idx] = partial(Recorder, add_time=True, silent=True) |
|
|
return self.learn |
|
|
|
|
|
def __exit__(self, *args): |
|
|
fastai.basic_train.master_bar,fastai.basic_train.progress_bar = master_bar,progress_bar |
|
|
self.learn.callback_fns = self.orig_callback_fns |
|
|
|
