config.py

import os
import torch

# Lambdas for loss weighting
LAMBDA_ADV = 1
LAMBDA_FEAT = 100
LAMBDA_REC = 1
lambdas = [LAMBDA_ADV, LAMBDA_FEAT, LAMBDA_REC]

# Learning params
STRIDES = [8, 5, 4, 2]
TENSOR_CUT = 64000
N_EPOCHS = 15
N_WARMUP_EPOCHS = 3
TRAIN_DISC_EVERY = 2
BATCH_SIZE = 6
LR = 1e-10
SAVE_FOLDER = os.path.join("/home/woody/iwi1/iwi1010h/checkpoints/SoundStream/", os.environ['SLURM_JOBID'])
SAVE_ACTIVATIONS = SAVE_FOLDER

# Windows length and hop for stft
W, H = 1024, 256
SR = 24000

# Data names that should be at WORK
TRAIN_FILE = "train-clean-360.tar.gz"
TEST_FILE = "test-clean.tar.gz"

# If continue training
RESUME = True
CHECKPOINT_REPOSITORY = os.path.join("/home/woody/iwi1/iwi1010h/checkpoints/SoundStream/", "739113")
CHECKPOINT_NAME = "la_1_lr_1_lf_100_ep_11_01-08_18:38.cpt"

DEVICE = str(torch.device("cuda" if torch.cuda.is_available() else "cpu"))