Convert Legacy to Trainer YAML

Learn how to convert legacy YAMLs into the new Trainer YAML configuration.

Converter Tool

The YAML specification changed after version 2.2 to make use of the Trainer class. You can use the converter tool to convert any legacy YAML to work with later versions:

ModelZoo CLI

cszoo config convert_legacy /path/to/v1.yaml -o /path/to/v2.yaml

Parameter Index

If you have a Legacy YAML specification and want to find out how to specify a specific parameter in the Trainer YAML specification, please use the tabs below to find the Legacy and Trainer specifications for each parameter.

eval_input

Legacy Spec

eval_input:
  ...

eval_input.micro_batch_size

Legacy Spec

eval_input:
  micro_batch_size: ...

model

Legacy Spec

model:
  ...

model.compression

Legacy Spec

model:
  compression:
    ...

model.fp16_type

Legacy Spec

model:
  fp16_type: ...

model.lora_params

Legacy Spec

model:
  lora_params: ...

model.mixed_precision

Legacy Spec

model:
  mixed_precision: ...

model.selective_grad

Legacy Spec

model:
  selective_grad ...

optimizer

Legacy Spec

optimizer:
  ...

optimizer.grad_accum_steps

Legacy Spec

optimizer:
  grad_accum_steps: ...

optimizer.initial_loss_scale

Legacy Spec

optimizer:
  initial_loss_scale: ...

optimizer.learning_rate

Legacy Spec

optimizer:
  learning_rate:
    ...

optimizer.log_summaries

Legacy Spec

optimizer:
  log_summaries: ...

optimizer.loss_scaling_factor

Legacy Spec

optimizer:
  loss_scaling_factor: ...

optimizer.max_gradient_norm

Legacy Spec

optimizer:
  max_gradient_norm: ...

optimizer.max_gradient_value

Legacy Spec

optimizer:
  max_gradient_value: ...

optimizer.max_loss_scale

Legacy Spec

optimizer:
  max_loss_scale: ...

optimizer.min_loss_scale

Legacy Spec

optimizer:
  min_loss_scale: ...

optimizer.steps_per_increase

Legacy Spec

optimizer:
  steps_per_increase: ...

runconfig.act_memory_gi

Legacy Spec

runconfig:
  act_memory_gi: ...

runconfig.autoload_last_checkpoint

Legacy Spec

runconfig:
  autoload_last_checkpoint: ...

runconfig.check_loss_values

Legacy Spec

runconfig:
  check_loss_values: ...

runconfig.checkpoint_path

Legacy Spec

runconfig:
  checkpoint_path: ...

runconfig.checkpoint_steps

Legacy Spec

runconfig:
  checkpoint_steps: ...

runconfig.cmd_memory_gi

Legacy Spec

runconfig:
  cmd_memory_gi: ...

runconfig.compile_crd_memory_gi

Legacy Spec

runconfig:
  compile_crd_memory_gi: ...

runconfig.compile_dir

Legacy Spec

runconfig:
  compile_dir: ...

runconfig.compile_only

Legacy Spec

runconfig:
  compile_only: ...

runconfig.credentials_path

Legacy Spec

runconfig:
  credentials_path: ...

runconfig.debug_args

Legacy Spec

runconfig:
  debug_args:
    ...

runconfig.debug_args_path

Legacy Spec

runconfig:
  debug_args_path: ...

runconfig.disable_strict_checkpoint_loading

Legacy Spec

runconfig:
  disable_strict_checkpoint_loading: ...

runconfig.disable_version_check

Legacy Spec

runconfig:
  disable_version_check: ...

runconfig.dist_backend

Legacy Spec

runconfig:
  dist_backend: ...

runconfig.drop_data

Legacy Spec

runconfig:
  drop_data: ...

runconfig.dump_activations

Legacy Spec

runconfig:
  dump_activations: ...

runconfig.enable_act_frequency

Legacy Spec

runconfig:
  enable_act_frequency: ...

runconfig.enable_distributed

Legacy Spec

runconfig:
  enable_distributed: ...

runconfig.eval_frequency

Legacy Spec

runconfig:
  eval_frequency: ...

runconfig.eval_steps

Legacy Spec

runconfig:
  eval_steps: ...

runconfig.execute_crd_memory_gi

Legacy Spec

runconfig:
  execute_crd_memory_gi: ...

runconfig.experimental.listeners

Legacy Spec

runconfig:
  experimental:
    listeners:
      ...

runconfig.init_method

Legacy Spec

runconfig:
  init_method: ...

runconfig.job_labels

Legacy Spec

runconfig:
  job_labels:
    ...

runconfig.job_priority

Legacy Spec

runconfig:
  job_priority: ...

runconfig.job_time_sec

Legacy Spec

runconfig:
  job_time_sec: ...

runconfig.lazy_initialization

Legacy Spec

runconfig:
  lazy_initialization: ...

runconfig.load_checkpoint_states

Legacy Spec

runconfig:
  load_checkpoint_states: ...

runconfig.log_initialization

Legacy Spec

runconfig:
  log_initialization: ...

runconfig.log_input_summaries

Legacy Spec

runconfig:
  log_input_summaries: ...

runconfig.log_steps

Legacy Spec

runconfig:
  log_steps: ...

runconfig.logging

Legacy Spec

runconfig:
  logging: ...

runconfig.main_process_id

Legacy Spec

runconfig:
  main_process_id: ...

runconfig.max_checkpoints

Legacy Spec

runconfig:
  max_checkpoints: ...

runconfig.max_steps

Legacy Spec

runconfig:
  max_steps: ...

runconfig.mgmt_address

Legacy Spec

runconfig:
  mgmt_address: ...

runconfig.mgmt_namespace

Legacy Spec

runconfig:
  mgmt_namespace: ...

runconfig.model_dir

Legacy Spec

runconfig:
  model_dir: ...

runconfig.mount_dirs

Legacy Spec

runconfig:
  mount_dirs:
    ...

runconfig.num_act_servers

Legacy Spec

runconfig:
  num_act_servers: ...

runconfig.num_csx

Legacy Spec

runconfig:
  num_csx: ...

runconfig.num_epochs

Legacy Spec

runconfig:
  num_epochs: ...

runconfig.num_steps

Legacy Spec

runconfig:
  num_steps: ...

runconfig.num_wgt_servers

Legacy Spec

runconfig:
  num_wgt_servers: ...

runconfig.num_workers_per_csx

Legacy Spec

runconfig:
  num_workers_per_csx: ...

runconfig.op_profiler_config

Legacy Spec

runconfig:
  op_profiler_config:
    ...

runconfig.precision_opt_level

Legacy Spec

runconfig:
  precision_opt_level: ...

runconfig.python_paths

Legacy Spec

runconfig:
  python_paths:
    ...

runconfig.retrace_every_iteration

Legacy Spec

runconfig:
  retrace_every_iteration: ...

runconfig.save_initial_checkpoint

Legacy Spec

runconfig:
  save_initial_checkpoint: ...

runconfig.seed

Legacy Spec

runconfig:
  seed: ...

runconfig.steps_per_epoch

Legacy Spec

runconfig:
  steps_per_epoch: ...

runconfig.sync_batchnorm

Legacy Spec

runconfig:
  sync_batchnorm: ...

runconfig.target_device

Legacy Spec

runconfig:
  target_device: ...

runconfig.transfer_processes

Legacy Spec

runconfig:
  transfer_processes: ...

runconfig.validate_only

Legacy Spec

runconfig:
  validate_only: ...

runconfig.wgt_memory_gi

Legacy Spec

runconfig:
  wgt_memory_gi: ...

runconfig.wrk_memory_gi

Legacy Spec

runconfig:
  wrk_memory_gi: ...

runconfig.wsc_log_level

Legacy Spec

runconfig:
  wsc_log_level:
    ...

sparsity

Legacy Spec

sparsity:
  ...

sparsity.add_summaries

Legacy Spec

sparsity:
  add_summaries: ...

train_input

Legacy Spec

train_input:
  ...

train_input.micro_batch_size

Legacy Spec

train_input:
  micro_batch_size: ...

wandb.group

Legacy Spec

wandb:
  group: ...

wandb.job_type

Legacy Spec

wandb:
  job_type: ...

wandb.project

Legacy Spec

wandb:
  project: ...

wandb.resume

Legacy Spec

wandb:
  resume: ...

wandb.run_id

Legacy Spec

wandb:
  run_id: ...

wandb.run_name

Legacy Spec

wandb:
  run_name: ...

wandb.tags

Legacy Spec

wandb:
  tags: ...

Assistant

Responses are generated using AI and may contain mistakes.