config/config.yaml

# WARNING: This is the base configuration file shared across ALL solvers in AudioCraft
# Please don't update this file directly. Instead use distinct configuration files
# to override the below configuration.
defaults:
  - _self_
  - dset: default
  - solver: default

device: cuda
dtype: float32
autocast: false
autocast_dtype: bfloat16
seed: 2036
show: false  # just show the model and its size and exit
continue_from:  # continue from a given sig or path
execute_only:  # can be set to generate/evaluate/valid to run that stage
execute_inplace: false # don't enforce continue_from to be set
                       # to enable inplace execution of the stage. This assume
                       # that you know what you are doing and execute stage
                       # preserving the original xp sig.
benchmark_no_load: false  # if set to true, will repeat the same batch instead of loading them

efficient_attention_backend: torch  # can be torch or xformers.
num_threads: 1                      # called with torch.set_num_thread.
mp_start_method: forkserver               # multiprocessing method (spawn, fork or fork_server).


label:  # use this if you want twice the same exp, with a name.

# logging parameters
logging:
  level: INFO
  log_updates: 10
  log_tensorboard: false
  log_wandb: false
tensorboard:
  with_media_logging: false
  name:  # optional name for the experiment
  sub_dir:  # optional sub directory to store tensorboard data
wandb:
  with_media_logging: true
  project:  # project name
  name:  # optional name for the experiment
  group:  # optional group

# SLURM launcher configuration.
slurm:
  gpus: 4  # convenience parameter, number of GPUs to use.
  mem_per_gpu: 40  # in GB, total mem is automatically scaled with `gpus`.
  time: 3600
  constraint:
  partition:
  comment:
  setup: []
  exclude: ''

# dora parameters
dora:
  # Output folder for all artifacts of an experiment.
  dir: /checkpoint/${oc.env:USER}/experiments/audiocraft/outputs
  # The following entries will be ignored by dora when computing the unique XP signature.
  # Note that slurm.* and dora.* are automatically ignored.
  exclude: [
    'device', 'wandb.*', 'tensorboard.*', 'logging.*',
    'dataset.num_workers', 'eval.num_workers', 'special.*',
    'metrics.visqol.bin', 'metrics.fad.bin',
    'execute_only', 'execute_best', 'generate.every',
    'optim.eager_sync', 'profiler.*', 'deadlock.*',
    'efficient_attention_backend', 'num_threads', 'mp_start_method',
  ]
  use_rendezvous: false
  # for grids, always run from a clean repo, allowing reliable runs and storing
  # the exact commit. Your repo must be absolutely pristine clean.
  # Local `dora run` are not impacted for easier debugging.
  git_save: true