dvc.lock

schema: '2.0'
stages:
  train:
    cmd: python3 source/train.py
    deps:
    - path: data/processed/
      hash: md5
      md5: 64793a00435610abbcd24e02ea03773f.dir
      size: 743300167
      nfiles: 3
    - path: source/network/
      hash: md5
      md5: 942bd26cda89270c31ff21160032bc27.dir
      size: 32837
      nfiles: 8
    - path: source/network/dataset.py
      hash: md5
      md5: 6a33e657139e21b977f4d2deb79bf1e4
      size: 4231
    - path: source/network/metrics.py
      hash: md5
      md5: edf8322a9643f164e7382b8a38ef00ae
      size: 1162
    - path: source/network/ravepqmf.py
      hash: md5
      md5: ff004d5afed02958b85ef7f204a0545f
      size: 6965
    - path: source/network/tcn.py
      hash: md5
      md5: 865b60460ae2b59dfbeba6517788e12d
      size: 2163
    - path: source/network/testing.py
      hash: md5
      md5: 1cf9d854e6a32543876daf71a03c7b6d
      size: 3701
    - path: source/network/training.py
      hash: md5
      md5: ec1ccb91860e86a33d31c98bd7fd39bf
      size: 8642
    - path: source/train.py
      hash: md5
      md5: daa0fbad12ff76b65ba3c06b920a759b
      size: 7251
    params:
      params.yaml:
        general.input_size: 524288
        general.random_seed: 55
        general.sample_rate: 44100
        train.batch_size: 32
        train.device_request: cuda
        train.dilation_growth: 9
        train.epochs: 1
        train.input_file: data/processed/preprocessed_data.pt
        train.kernel_size: 32
        train.latent_dim: 128
        train.lr: 0.001
        train.n_bands: 16
        train.n_blocks: 3
        train.n_channels: 32
        train.n_inputs: 1
        train.name: model
        train.scheduler_rate: 0.1
        train.use_kl: false
        train.use_skip: true
        train.use_tcn: false