File size: 694 Bytes
2469f83 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
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
#20230109-181938
#The training is starting
#20230111-021334
#The training is ending
|