class_docstr: Check that loading a checkpoint during training works
  Check that loading a checkpoint for benchmarking works
class_docstr_tok: ["Check","that","loading","a","checkpoint","during","training","works","Check","that","loading","a","checkpoint","for","benchmarking","works"]
code: def test_restart_after_preemption_at_iteration(self) -> None:
      with in_temporary_directory():
          config = self._create_pretraining_config(with_fsdp=False)
          config.CHECKPOINT.CHECKPOINT_ITER_FREQUENCY = 3
          # TODO - understand why the losses do not match exactly for iteration preemption
          self.run_preemption_test(config, compare_losses=False)
  
imports: import os
  import unittest
  from vissl.config.attr_dict import AttrDict
  from vissl.utils.checkpoint import CheckpointFormatConverter
  from vissl.utils.hydra_config import compose_hydra_configuration, convert_to_attrdict
  from vissl.utils.test_utils import gpu_test, in_temporary_directory, run_integration_test
label_desc: Use image node_id 5 for calling the TestStateCheckpointing obj's underlying member method code with example usage: obj.test_restart_after_preemption_at_iteration() without return types