如果设置
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.

    dwmatrix = create_downsample_fn_matrix(self.num_branches, features, self.num_branches, output_channels)

should be
dwmatrix = create_downsample_fn_matrix(self.num_branches, features, self.num_outputs, output_channels)