Method basename

tensorpack/graph_builder/training.py:64–65

(x)

        nvars = [len(k) for k in grad_list]

        def basename(x):
            return re.sub('tower[0-9]+/', '', x.op.name)

        if len(set(nvars)) != 1:
            names_per_gpu = [{basename(k[1]) for k in grad_and_vars} for grad_and_vars in grad_list]