cfg = {
  'MODEL_SAVE_PATH': './saved_models/',

  'USE_CUDA': False,

  'TRAIN_BATCH_SIZE': 256,

  'TEST_BATCH_SIZE': 1000,

  'TRAIN_EPOCHS': 30,

  'NUM_COUPLING_LAYERS': 4,

  'NUM_NET_LAYERS': 6,  # neural net layers for each coupling layer

  'NUM_HIDDEN_UNITS': 1000
}
