56 lines
No EOL
1.1 KiB
INI
56 lines
No EOL
1.1 KiB
INI
|
|
[DEFAULTS]
|
|
|
|
#name of the run
|
|
name = stable_audio_tools
|
|
|
|
# the batch size
|
|
batch_size = 8
|
|
|
|
# number of GPUs to use for training
|
|
num_gpus = 1
|
|
|
|
# number of nodes to use for training
|
|
num_nodes = 1
|
|
|
|
# Multi-GPU strategy for PyTorch Lightning
|
|
strategy = ""
|
|
|
|
# Precision to use for training
|
|
precision = "16-mixed"
|
|
|
|
# number of CPU workers for the DataLoader
|
|
num_workers = 8
|
|
|
|
# the random seed
|
|
seed = 42
|
|
|
|
# Batches for gradient accumulation
|
|
accum_batches = 1
|
|
|
|
# Number of steps between checkpoints
|
|
checkpoint_every = 10000
|
|
|
|
# trainer checkpoint file to restart training from
|
|
ckpt_path = ''
|
|
|
|
# model checkpoint file to start a new training run from
|
|
pretrained_ckpt_path = ''
|
|
|
|
# Checkpoint path for the pretransform model if needed
|
|
pretransform_ckpt_path = ''
|
|
|
|
# configuration model specifying model hyperparameters
|
|
model_config = ''
|
|
|
|
# configuration for datasets
|
|
dataset_config = ''
|
|
|
|
# directory to save the checkpoints in
|
|
save_dir = ''
|
|
|
|
# gradient_clip_val passed into PyTorch Lightning Trainer
|
|
gradient_clip_val = 0.0
|
|
|
|
# remove the weight norm from the pretransform model
|
|
remove_pretransform_weight_norm = '' |