[Arguments]

port = 8097
env = main
# resume = ./records/Gaussian/HSIC-250-12/0.01_sigma 0.01/Save/GaussianHSIC-epoch=159-val_loss=0.622.ckpt
same_env = Yes
save_results = No

# project options
project_name=ARL
monitor={"metric": "loss", "decreasing": 1}

# dataset options
dataset=Gaussian
dataset_type=GaussianPerp
dataset_options={"nsamples": 4000}

# model options
batch_size_test=2000
batch_size_train=500
model_type = EncDecGaussian
model_adv_type = AdvGaussian
model_options = {"ndim":1, "nout":2, "r":1, "hdl":4}
r = 1
control_type = ARLReg
# control_options = {"type":"conditional"}

kernel_type = GaussianKernel
kernel_labels = yes
# kernel_options={"sigma": 0.1}
sigma_x = 1
sigma_y = 1
sigma_s = 1


# loss options
loss_type = Regression
loss_options={}
evaluation_type = MeanSquaredError
evaluation_options={}
tau = .0

fairness_type = DEP
fairness_options = {}

manual_seed = 0
nepochs = 500
check_val_every_n_epochs = 50
control_epoch = 0

optim_method = AdamW
learning_rate = 3e-4

# optim_options = {"momentum": 0.9, "weight_decay": 2e-5}
optim_options = {"weight_decay": 2e-5}


scheduler_method = StepLR
scheduler_options = {"step_size": 500, "gamma": 0.5}

#scheduler_method = CosineAnnealingLR
#scheduler_options = {"T_max": 5}

# cpu/gpu settings
ngpu = 1
nthreads = 4