Function init_print

accessory/model/multi_gpu_wrapper.py:26–35

(is_master)

]

def init_print(is_master):
    builtin_print = builtins.print
    def _print(*args, **kwargs):
        force = kwargs.pop('force', False)
        kwargs['flush'] = True # flush should always be true
        if is_master or force:
            builtin_print('[model process] ', end='')  # print with time stamp
            builtin_print(*args, **kwargs)

    builtins.print = _print

class ResetException(Exception):
    pass