code: def log_to_screen(self):
      # log to screen
      if self.phase == RunPhase.TRAIN:
          # for the training phase - we log during the episode to visualize the progress in training
          log = OrderedDict()
          if self.task_id is not None:
              log["Worker"] = self.task_id
          log["Episode"] = self.current_episode
          log["Loss"] = self.loss.values[-1]
          log["Training iteration"] = self.training_iteration
          screen.log_dict(log, prefix="Training")
      else:
          # for the evaluation phase - logging as in regular RL
          super().log_to_screen()
  
file_path: coach/rl_coach/agents/imitation_agent.py
imports: from collections import OrderedDict
  from typing import Union
  from rl_coach.agents.agent import Agent
  from rl_coach.core_types import RunPhase, ActionInfo
  from rl_coach.logger import screen
  from rl_coach.spaces import DiscreteActionSpace
label_desc: Use image node_id 5 for calling the ImitationAgent obj's underlying member method code with example usage: obj.log_to_screen() without return types