Guanxiong commited on
Commit
5cfabeb
1 Parent(s): aaa9378

add cplx_small model

Browse files
PixelCNN/cplx_small/config.yaml ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model: PIXELCNN
2
+ batch_size: 6
3
+ input_shape:
4
+ - 256
5
+ - 256
6
+ - 2
7
+ nr_resnet: 3
8
+ nr_filters: 128
9
+ nr_logistic_mix: 10
10
+ data_chns: 2
11
+ dropout_rate: 0.5
12
+ itg_interval: 255.0
13
+ rlt: 1
14
+ layer_norm: false
15
+ conditional: false
16
+ lr_warm_up_steps: 1000
17
+ lr_start: 0.0001
18
+ lr_min: 0.0001
19
+ lr_max: 0.0001
20
+ lr_max_decay_steps: 2000
21
+ max_keep: 100
22
+ max_epochs: 500
23
+ save_interval: 50
24
+ saved_name: pixelcnn
25
+ log_folder: /home/gluo/workspace/nlinv_prior/logs
26
+ num_prepare: 10
27
+ print_loss: true
28
+ train_list: /home/gluo/workspace/nlinv_prior/data/hku/hku_train
29
+ test_list: /home/gluo/workspace/nlinv_prior/data/hku/hku_test
30
+ nr_gpu: 4
31
+ gpu_id: 0,1,2,3
32
+ #20230114-164131
33
+ #The training is starting
34
+ #20230114-230037
35
+ #The training is ending
PixelCNN/cplx_small/pixelcnn_500.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e36de8ae13325bcc1743693f21f95bd4d329b668a11cffdfa72f6479a1f75622
3
+ size 260950952
PixelCNN/cplx_small/pixelcnn_500.index ADDED
Binary file (37 kB). View file
 
PixelCNN/cplx_small/pixelcnn_500.meta ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b050c207575350b6b6d284430e03c9f1da68d1e5203e52b47626af6a7650fe6b
3
+ size 17665457