CoolFace
Apppublic

ZeroTwo3/videoshop-backend

sourceHugging Faceopenrailupdated 3y agoView on Hugging Face
2likes
config.yaml76 linesDownload Raw Back to config
1# WARNING: This is the base configuration file shared across ALL solvers in AudioCraft2# Please don't update this file directly. Instead use distinct configuration files3# to override the below configuration.4defaults:5  - _self_6  - dset: default7  - solver: default8 9device: cuda10dtype: float3211autocast: false12autocast_dtype: bfloat1613seed: 203614show: false  # just show the model and its size and exit15continue_from:  # continue from a given sig or path16execute_only:  # can be set to generate/evaluate/valid to run that stage17execute_inplace: false # don't enforce continue_from to be set18                       # to enable inplace execution of the stage. This assume19                       # that you know what you are doing and execute stage20                       # preserving the original xp sig.21benchmark_no_load: false  # if set to true, will repeat the same batch instead of loading them22 23efficient_attention_backend: torch  # can be torch or xformers.24num_threads: 1                      # called with torch.set_num_thread.25mp_start_method: forkserver               # multiprocessing method (spawn, fork or fork_server).26 27 28label:  # use this if you want twice the same exp, with a name.29 30# logging parameters31logging:32  level: INFO33  log_updates: 1034  log_tensorboard: false35  log_wandb: false36tensorboard:37  with_media_logging: false38  name:  # optional name for the experiment39  sub_dir:  # optional sub directory to store tensorboard data40wandb:41  with_media_logging: true42  project:  # project name43  name:  # optional name for the experiment44  group:  # optional group45 46# SLURM launcher configuration.47slurm:48  gpus: 4  # convenience parameter, number of GPUs to use.49  mem_per_gpu: 40  # in GB, total mem is automatically scaled with `gpus`.50  time: 360051  constraint:52  partition:53  comment:54  setup: []55  exclude: ''56 57# dora parameters58dora:59  # Output folder for all artifacts of an experiment.60  dir: /checkpoint/${oc.env:USER}/experiments/audiocraft/outputs61  # The following entries will be ignored by dora when computing the unique XP signature.62  # Note that slurm.* and dora.* are automatically ignored.63  exclude: [64    'device', 'wandb.*', 'tensorboard.*', 'logging.*',65    'dataset.num_workers', 'eval.num_workers', 'special.*',66    'metrics.visqol.bin', 'metrics.fad.bin',67    'execute_only', 'execute_best', 'generate.every',68    'optim.eager_sync', 'profiler.*', 'deadlock.*',69    'efficient_attention_backend', 'num_threads', 'mp_start_method',70  ]71  use_rendezvous: false72  # for grids, always run from a clean repo, allowing reliable runs and storing73  # the exact commit. Your repo must be absolutely pristine clean.74  # Local `dora run` are not impacted for easier debugging.75  git_save: true76