question: Control log_dict's on_epoch log name
context: Hey!
  I tried to change to using the new way of logging through MetricCollection and self.log_dict instead of logging every metric through self.log on training step and test_epoch_end. However, each metric is then logged as [metric_name]_epoch_[epoch_number] which creates a new graph for every epoch instead of allowing me to use epoch on the x-axis of my graphs (on Comet, if that is relevant). Is there a way to control this behaviour of log_dict, or do I just have to keep logging "manually" to control log name?
answer: Hi @FluidSense,
  Could you try just updating the MetricCollection during the _step method and then log in _epoch_end method. Something like:
  def training_step(self, batch, batch_idx):
      logits = self(x)
      self.train_metrics.update(logits, y)
  
  def train_epoch_end(self, outputs):
      self.log_dict(self.train_metrics.compute())