PlutoQyl

You can add this in runner： def _write_metrics(self, loss_dict, data_time: float, prefix: str = ""): super(DefaultRunner, self)._write_metrics(loss_dict, data_time, prefix) if comm.is_main_process() and self.iter != 0: writer: TensorboardXWriter = self.get_writer(writer_type="tensorboard") writer.write_angular_update(self.optimizer)

You can use this code instead. semi_sampler.py GroupSemiBalanceSampler `shuffled_indice_per_dataset[j] = np.concatenate( ( shuffled_indice_per_dataset[j], # np.random.shuffle(indice_per_dataset[j]), indice_per_dataset[j][ list( np.random.permutation( int(indice_per_dataset[j].shape[0]), ) ) ], ) )`

Have you tried centerness branch？

> Thanks! I will take a look at your source code for simpletuner, and see if that helps me understand how to do it. I'm still trying to wrap my...