|
model: PIXELCNN |
|
batch_size: 6 |
|
input_shape: |
|
- 256 |
|
- 256 |
|
- 2 |
|
nr_resnet: 3 |
|
nr_filters: 128 |
|
nr_logistic_mix: 10 |
|
data_chns: 2 |
|
dropout_rate: 0.5 |
|
itg_interval: 255.0 |
|
rlt: 1 |
|
layer_norm: false |
|
conditional: false |
|
lr_warm_up_steps: 1000 |
|
lr_start: 0.0001 |
|
lr_min: 0.0001 |
|
lr_max: 0.0001 |
|
lr_max_decay_steps: 2000 |
|
max_keep: 100 |
|
max_epochs: 100 |
|
save_interval: 10 |
|
saved_name: pixelcnn |
|
log_folder: /home/gluo/workspace/nlinv/logs |
|
num_prepare: 10 |
|
print_loss: true |
|
train_list: /home/gluo/workspace/nlinv/scripts/data/abide/abide_train_p |
|
test_list: /home/gluo/workspace/nlinv/scripts/data/abide/abide_test_p |
|
nr_gpu: 3 |
|
gpu_id: 0,1,2 |
|
|
|
|
|
|
|
|
|
|
