code: def on_step_begin(self, args, state, control, **kwargs):
      if state.global_step == 0:
          control.should_log = True
          control.should_evaluate = True
  
file_path: happy-transformer/happytransformer/fine_tuning_util.py
imports: from dataclasses import dataclass
  from typing import Union
  from datasets import Dataset
  from transformers import PreTrainedTokenizer, TrainerCallback
  from happytransformer.args import GENTrainArgs, WPTrainArgs, GENEvalArgs, WPEvalArgs
label_desc: Use image node_id 1 for calling the FistStep obj's underlying member method code with example usage: obj.on_step_begin(args, state, control) without return types