Skip to contents

Train model locally using available hardware

Usage

prepare_training_config(
  data_dir,
  dataset_id,
  dataset_version,
  model_id,
  num_classes,
  backbone,
  freeze_at,
  max_iter,
  learning_rate,
  device,
  eval_period,
  checkpoint_period,
  ims_per_batch,
  num_workers,
  hpc_cpus_per_task,
  hpc_mem,
  gpus
)