Exception[pytorch_lightning]: You are trying to `self.log()` but it is not managed by the `Trainer` control flow about fots.pytorch HOT 1 CLOSED

trainer = Trainer(
    logger=wandb_logger,
    callbacks=[checkpoint_callback],
    max_epochs=config.trainer.epochs,
    default_root_dir=root_dir,
    gpus=gpus,
    # accelerator='ddp',
    benchmark=True,
    sync_batchnorm=True,
    precision=config.precision,
    # log_gpu_memory=config.trainer.log_gpu_memory,
    log_every_n_steps=config.trainer.log_every_n_steps,
    overfit_batches=config.trainer.overfit_batches,
    weights_summary='full',
    terminate_on_nan=config.trainer.terminate_on_nan,
    fast_dev_run=config.trainer.fast_dev_run,
    check_val_every_n_epoch=config.trainer.check_val_every_n_epoch,
    resume_from_checkpoint=resume_ckpt)