how to sync LightningModule.log value, to let EarlyStoping, ModelCheckpoint callbacks work correctly?

[peiliu0408]

(pytorch_lightning 1.7.6, 2 GPUs, the validation metric (val_map) is got in validation_epoch_end on cpu mode.

` def validation_epoch_end(self, results):

results = collect_results() # work correctly
if self.trainer.is_global_rank():
    map = get_metrics(results) # on cpu, and work correctly
    self.log('val_ap', map, rank_zero_only=True) `

errors: the ModelCheck(monitor='val_ap') could not find the monitored key in the returned metrics: ['step', 'epoch'].

how can I to let model only log on rank 0 after each validation epoch, and still can compatible other callbacks?

thanks.