Remove mlperf logging

Signed-off-by: Kunjan patel <kunjanp@google.com>

Author: coolkp

src/maxdiffusion/mllog_utils.py

@@ -21,7 +21,6 @@
 from maxdiffusion import (
     max_logging,
     pyconfig,
-    mllog_utils,
 )
 
 from maxdiffusion.train_utils import (
@@ -39,7 +38,6 @@ def train(config):
 def main(argv: Sequence[str]) -> None:
   pyconfig.initialize(argv)
   config = pyconfig.config
-  mllog_utils.train_init_start(config)
   validate_train_config(config)
   max_logging.log(f"Found {jax.device_count()} devices.")
   train(config)

src/maxdiffusion/train_flux.py

@@ -21,7 +21,6 @@
 from maxdiffusion import (
     max_logging,
     pyconfig,
-    mllog_utils,
 )
 
 from maxdiffusion.trainers.sdxl_trainer import StableDiffusionXLTrainer
@@ -39,7 +38,6 @@ def train(config):
 def main(argv: Sequence[str]) -> None:
   pyconfig.initialize(argv)
   config = pyconfig.config
-  mllog_utils.train_init_start(config)
   validate_train_config(config)
   max_logging.log(f"Found {jax.device_count()} devices.")
   train(config)