from the example of pytorch example in Katib, it can collect the metric for loss and accuracy. I tried to calculate other measurements during train() and wanted to be collect to display in katib(AutoML) ui, I already put in logging.info() and also put it in spec.objective.additionalMetricNames in yaml file, but it didn't work.