code: def test_training(self):
      # Test an Apex AMP training
      config = get_fast_test_task_config()
      config["amp_args"] = {"opt_level": "O2"}
      task = build_task(config)
      task.set_use_gpu(True)
      trainer = LocalTrainer()
      trainer.train(task)
  
      # Test a Pytorch AMP training
      config["amp_args"] = {"amp_type": "pytorch"}
      task = build_task(config)
      task.set_use_gpu(True)
      trainer = LocalTrainer()
      trainer.train(task)
  
file_path: ClassyVision/test/manual/tasks_classification_task_amp_test.py
imports: import copy
  import unittest
  import torch
  from classy_vision.tasks import build_task, ClassificationTask
  from classy_vision.trainer import LocalTrainer
  from test.generic.config_utils import get_fast_test_task_config, get_test_task_config
label_desc: Use image node_id 2 for calling the TestClassificationTaskAMP obj's underlying member method code with example usage: obj.test_training() without return types