crlandsc's picture
hi hat config
6928044
seed: 12345
train: true
ignore_warnings: true
print_config: false
work_dir: ${hydra:runtime.cwd}
logs_dir: ${work_dir}${oc.env:DIR_LOGS}
data_dir: ${work_dir}${oc.env:DIR_DATA}
ckpt_dir: ${logs_dir}/runs/${now:%Y-%m-%d-%H-%M-%S}
module: main.module_base
batch_size: 16
accumulate_grad_batches: 2
num_workers: 16
sampling_rate: 44100
length: 32768
channels: 2
log_every_n_steps: 1000
model:
_target_: ${module}.Model
lr: 0.0001
lr_beta1: 0.95
lr_beta2: 0.999
lr_eps: 1.0e-06
lr_weight_decay: 0.001
ema_beta: 0.995
ema_power: 0.7
model:
_target_: main.DiffusionModel
net_t:
_target_: ${module}.UNetT
in_channels: 2
channels:
- 32
- 32
- 64
- 64
- 128
- 128
- 256
- 256
factors:
- 1
- 2
- 2
- 2
- 2
- 2
- 2
- 2
items:
- 2
- 2
- 2
- 2
- 2
- 2
- 4
- 4
attentions:
- 0
- 0
- 0
- 0
- 0
- 1
- 1
- 1
attention_heads: 8
attention_features: 64
datamodule:
_target_: main.module_base.Datamodule
dataset:
_target_: audio_data_pytorch.WAVDataset
path: ./data/kicks
recursive: true
sample_rate: ${sampling_rate}
transforms:
_target_: audio_data_pytorch.AllTransform
crop_size: ${length}
stereo: true
source_rate: ${sampling_rate}
target_rate: ${sampling_rate}
loudness: -20
val_split: 0.05
batch_size: ${batch_size}
num_workers: ${num_workers}
pin_memory: true
callbacks:
rich_progress_bar:
_target_: pytorch_lightning.callbacks.RichProgressBar
model_checkpoint:
_target_: pytorch_lightning.callbacks.ModelCheckpoint
monitor: valid_loss
save_top_k: 1
save_last: true
mode: min
verbose: false
dirpath: ${logs_dir}/ckpts/${now:%Y-%m-%d-%H-%M-%S}
filename: '{epoch:02d}-{valid_loss:.3f}'
model_summary:
_target_: pytorch_lightning.callbacks.RichModelSummary
max_depth: 2
audio_samples_logger:
_target_: main.module_base.SampleLogger
num_items: 4
channels: ${channels}
sampling_rate: ${sampling_rate}
length: ${length}
sampling_steps:
- 50
use_ema_model: true
loggers:
wandb:
_target_: pytorch_lightning.loggers.wandb.WandbLogger
project: ${oc.env:WANDB_PROJECT}
entity: ${oc.env:WANDB_ENTITY}
name: kicks_v7
job_type: train
group: ''
save_dir: ${logs_dir}
trainer:
_target_: pytorch_lightning.Trainer
gpus: 1
precision: 16
accelerator: gpu
min_epochs: 0
max_epochs: -1
enable_model_summary: false
log_every_n_steps: 1
check_val_every_n_epoch: null
val_check_interval: ${log_every_n_steps}
accumulate_grad_batches: ${accumulate_grad_batches}
ckpt: ./saved_models/kicks/kicks_v7_2.ckpt