code: def on_epoch_end(self, args, state, control, **kwargs):
      print(
          "GPU mem: Tot - ",
          torch.cuda.get_device_properties(0).total_memory,
          "res - ",
          torch.cuda.memory_reserved(0),
          "used - ",
          torch.cuda.memory_allocated(0),
      )
  
file_path: orion/examples/huggingface/main.py
imports: import logging
  import os
  import argparse
  from copy import deepcopy
  import hydra
  import numpy
  import torch
  from datasets import load_dataset, load_metric
  from omegaconf import DictConfig
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, DataCollatorForSeq2Seq, Seq2SeqTrainer, Seq2SeqTrainingArguments, TrainerCallback
label_desc: Use image node_id 1 for calling the GPUMemoryCallback obj's underlying member method code with example usage: obj.on_epoch_end(args, state, control) without return types