callbacks.py 文件源码

def __init__(self, trainer, dataset, start_epoch=0, momentum=0, batch_size=96):
        super(SemiSupervisedUpdater, self).__init__()

        self.trainer = trainer
        self.dataset = dataset
        self.start_epoch = start_epoch
        self.loader = DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=10, pin_memory=True)
        self.momentum = momentum