多进程训练时 cfg.NUM_TRAINERS 和 cfg.TRAINER_ID
Created by: jaffe-fly
如果设置
cfg.NUM_TRAINERS = 4
cfg.TRAINER_ID = 0,1,2,3
if self.shuffle and cfg.NUM_TRAINERS > 1: np.random.RandomState(self.shuffle_seed).shuffle(self.all_lines) num_lines = len(self.all_lines) // cfg.NUM_TRAINERS self.lines = self.all_lines[num_lines * cfg.TRAINER_ID: num_lines * (cfg.TRAINER_ID + 1)] self.shuffle_seed += 1
上面代码中的self.lines 时怎么取值的?