nutszebra_chainer.py 文件源码

def __init__(self, **kwargs):
        # initialization for chainer.Chain
        # If you don't initialize, model.to_gpu doesn't work, because no link
        super(Model, self).__init__(**kwargs)
        self.nz_save_model_epoch = 0
        self.nz_save_optimizer_epoch = 0
        self.nz_xp = self._check_cupy()
        self.nz_flag_computational_graph = False