CONFIG_NAME: 'stageI'

DATASET_NAME: 'RIR'
EMBEDDING_TYPE: 'cnn-rnn'
GPU_ID: '0,1'

DATA_DIR: '../data/Medium_Room'

EVAL_DIR: '../generate/embeddings/'
RIRSIZE: 4096
WORKERS: 4
STAGE: 1
TRAIN:
    FLAG: True
    BATCH_SIZE: 128
    MAX_EPOCH: 2000
    LR_DECAY_EPOCH: 40
    SNAPSHOT_INTERVAL: 50
    # DISCRIMINATOR_LR: 0.0002
    # GENERATOR_LR: 0.0002
    DISCRIMINATOR_LR: 0.00008
    GENERATOR_LR: 0.00008
    COEFF:
      KL: 2.0

GAN:
    CONDITION_DIM: 10
    DF_DIM: 96
    GF_DIM: 256

TEXT:
    DIMENSION: 10