| model: PIXELCNN | |
| batch_size: 6 | |
| input_shape: | |
| - 256 | |
| - 256 | |
| - 1 | |
| nr_resnet: 3 | |
| nr_filters: 128 | |
| nr_logistic_mix: 10 | |
| data_chns: 1 | |
| 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_prior/logs | |
| num_thread: 30 | |
| print_loss: true | |
| train_list: /home/gluo/workspace/nlinv_prior/data/abide/abide_mag_train | |
| test_list: /home/gluo/workspace/nlinv_prior/data/abide/abide_mag_test | |
| key: mag | |
| nr_gpu: 4 | |
| gpu_id: 0,1,2,3 | |