File size: 7,757 Bytes
2b5e26c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
7767517
119 132
Input  in0  0 1 in0
Split  dec_mid_rn1_split  1 2 in0 b1 b2
GroupNorm  dec_mid_rn1_norm1  1 1 b1 b3 0=32 1=512 2=1e-06 3=1
Swish  dec_mid_rn1_act1  1 1 b3 b4
Convolution  dec_mid_rn1_conv1  1 1 b4 b5 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_mid_rn1_norm2  1 1 b5 b6 0=32 1=512 2=1e-06 3=1
Swish  dec_mid_rn1_act2  1 1 b6 b7
Convolution  dec_mid_rn1_conv2  1 1 b7 b8 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_mid_rn1_add  2 1 b8 b2 b9 0=0
Split  dec_ub0_rn0_split  1 2 b9 b10 b11
GroupNorm  dec_ub0_rn0_norm1  1 1 b10 b12 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn0_act1  1 1 b12 b13
Convolution  dec_ub0_rn0_conv1  1 1 b13 b14 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub0_rn0_norm2  1 1 b14 b15 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn0_act2  1 1 b15 b16
Convolution  dec_ub0_rn0_conv2  1 1 b16 b17 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub0_rn0_add  2 1 b17 b11 b18 0=0
Split  dec_ub0_rn1_split  1 2 b18 b19 b20
GroupNorm  dec_ub0_rn1_norm1  1 1 b19 b21 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn1_act1  1 1 b21 b22
Convolution  dec_ub0_rn1_conv1  1 1 b22 b23 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub0_rn1_norm2  1 1 b23 b24 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn1_act2  1 1 b24 b25
Convolution  dec_ub0_rn1_conv2  1 1 b25 b26 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub0_rn1_add  2 1 b26 b20 b27 0=0
Split  dec_ub0_rn2_split  1 2 b27 b28 b29
GroupNorm  dec_ub0_rn2_norm1  1 1 b28 b30 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn2_act1  1 1 b30 b31
Convolution  dec_ub0_rn2_conv1  1 1 b31 b32 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub0_rn2_norm2  1 1 b32 b33 0=32 1=512 2=1e-06 3=1
Swish  dec_ub0_rn2_act2  1 1 b33 b34
Convolution  dec_ub0_rn2_conv2  1 1 b34 b35 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub0_rn2_add  2 1 b35 b29 b36 0=0
Convolution  dec_ub0_us_up1x1  1 1 b36 b37 0=2048 1=1 11=1 3=1 13=1 4=0 15=0 14=0 16=0 5=1 6=1048576
PixelShuffle  dec_ub0_us_ps  1 1 b37 b38 0=2 1=0
Convolution  dec_ub0_us_conv  1 1 b38 b39 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
Split  dec_ub1_rn0_split  1 2 b39 b40 b41
GroupNorm  dec_ub1_rn0_norm1  1 1 b40 b42 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn0_act1  1 1 b42 b43
Convolution  dec_ub1_rn0_conv1  1 1 b43 b44 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub1_rn0_norm2  1 1 b44 b45 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn0_act2  1 1 b45 b46
Convolution  dec_ub1_rn0_conv2  1 1 b46 b47 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub1_rn0_add  2 1 b47 b41 b48 0=0
Split  dec_ub1_rn1_split  1 2 b48 b49 b50
GroupNorm  dec_ub1_rn1_norm1  1 1 b49 b51 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn1_act1  1 1 b51 b52
Convolution  dec_ub1_rn1_conv1  1 1 b52 b53 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub1_rn1_norm2  1 1 b53 b54 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn1_act2  1 1 b54 b55
Convolution  dec_ub1_rn1_conv2  1 1 b55 b56 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub1_rn1_add  2 1 b56 b50 b57 0=0
Split  dec_ub1_rn2_split  1 2 b57 b58 b59
GroupNorm  dec_ub1_rn2_norm1  1 1 b58 b60 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn2_act1  1 1 b60 b61
Convolution  dec_ub1_rn2_conv1  1 1 b61 b62 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
GroupNorm  dec_ub1_rn2_norm2  1 1 b62 b63 0=32 1=512 2=1e-06 3=1
Swish  dec_ub1_rn2_act2  1 1 b63 b64
Convolution  dec_ub1_rn2_conv2  1 1 b64 b65 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
BinaryOp  dec_ub1_rn2_add  2 1 b65 b59 b66 0=0
Convolution  dec_ub1_us_up1x1  1 1 b66 b67 0=2048 1=1 11=1 3=1 13=1 4=0 15=0 14=0 16=0 5=1 6=1048576
PixelShuffle  dec_ub1_us_ps  1 1 b67 b68 0=2 1=0
Convolution  dec_ub1_us_conv  1 1 b68 b69 0=512 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=2359296
Split  dec_ub2_rn0_split  1 2 b69 b70 b71
GroupNorm  dec_ub2_rn0_norm1  1 1 b70 b72 0=32 1=512 2=1e-06 3=1
Swish  dec_ub2_rn0_act1  1 1 b72 b73
Convolution  dec_ub2_rn0_conv1  1 1 b73 b74 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=1179648
GroupNorm  dec_ub2_rn0_norm2  1 1 b74 b75 0=32 1=256 2=1e-06 3=1
Swish  dec_ub2_rn0_act2  1 1 b75 b76
Convolution  dec_ub2_rn0_conv2  1 1 b76 b77 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
Convolution  dec_ub2_rn0_shortcut  1 1 b71 b78 0=256 1=1 11=1 3=1 13=1 4=0 15=0 14=0 16=0 5=1 6=131072
BinaryOp  dec_ub2_rn0_add  2 1 b77 b78 b79 0=0
Split  dec_ub2_rn1_split  1 2 b79 b80 b81
GroupNorm  dec_ub2_rn1_norm1  1 1 b80 b82 0=32 1=256 2=1e-06 3=1
Swish  dec_ub2_rn1_act1  1 1 b82 b83
Convolution  dec_ub2_rn1_conv1  1 1 b83 b84 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
GroupNorm  dec_ub2_rn1_norm2  1 1 b84 b85 0=32 1=256 2=1e-06 3=1
Swish  dec_ub2_rn1_act2  1 1 b85 b86
Convolution  dec_ub2_rn1_conv2  1 1 b86 b87 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
BinaryOp  dec_ub2_rn1_add  2 1 b87 b81 b88 0=0
Split  dec_ub2_rn2_split  1 2 b88 b89 b90
GroupNorm  dec_ub2_rn2_norm1  1 1 b89 b91 0=32 1=256 2=1e-06 3=1
Swish  dec_ub2_rn2_act1  1 1 b91 b92
Convolution  dec_ub2_rn2_conv1  1 1 b92 b93 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
GroupNorm  dec_ub2_rn2_norm2  1 1 b93 b94 0=32 1=256 2=1e-06 3=1
Swish  dec_ub2_rn2_act2  1 1 b94 b95
Convolution  dec_ub2_rn2_conv2  1 1 b95 b96 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
BinaryOp  dec_ub2_rn2_add  2 1 b96 b90 b97 0=0
Convolution  dec_ub2_us_up1x1  1 1 b97 b98 0=1024 1=1 11=1 3=1 13=1 4=0 15=0 14=0 16=0 5=1 6=262144
PixelShuffle  dec_ub2_us_ps  1 1 b98 b99 0=2 1=0
Convolution  dec_ub2_us_conv  1 1 b99 b100 0=256 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=589824
Split  dec_ub3_rn0_split  1 2 b100 b101 b102
GroupNorm  dec_ub3_rn0_norm1  1 1 b101 b103 0=32 1=256 2=1e-06 3=1
Swish  dec_ub3_rn0_act1  1 1 b103 b104
Convolution  dec_ub3_rn0_conv1  1 1 b104 b105 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=294912
GroupNorm  dec_ub3_rn0_norm2  1 1 b105 b106 0=32 1=128 2=1e-06 3=1
Swish  dec_ub3_rn0_act2  1 1 b106 b107
Convolution  dec_ub3_rn0_conv2  1 1 b107 b108 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=147456
Convolution  dec_ub3_rn0_shortcut  1 1 b102 b109 0=128 1=1 11=1 3=1 13=1 4=0 15=0 14=0 16=0 5=1 6=32768
BinaryOp  dec_ub3_rn0_add  2 1 b108 b109 b110 0=0
Split  dec_ub3_rn1_split  1 2 b110 b111 b112
GroupNorm  dec_ub3_rn1_norm1  1 1 b111 b113 0=32 1=128 2=1e-06 3=1
Swish  dec_ub3_rn1_act1  1 1 b113 b114
Convolution  dec_ub3_rn1_conv1  1 1 b114 b115 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=147456
GroupNorm  dec_ub3_rn1_norm2  1 1 b115 b116 0=32 1=128 2=1e-06 3=1
Swish  dec_ub3_rn1_act2  1 1 b116 b117
Convolution  dec_ub3_rn1_conv2  1 1 b117 b118 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=147456
BinaryOp  dec_ub3_rn1_add  2 1 b118 b112 b119 0=0
Split  dec_ub3_rn2_split  1 2 b119 b120 b121
GroupNorm  dec_ub3_rn2_norm1  1 1 b120 b122 0=32 1=128 2=1e-06 3=1
Swish  dec_ub3_rn2_act1  1 1 b122 b123
Convolution  dec_ub3_rn2_conv1  1 1 b123 b124 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=147456
GroupNorm  dec_ub3_rn2_norm2  1 1 b124 b125 0=32 1=128 2=1e-06 3=1
Swish  dec_ub3_rn2_act2  1 1 b125 b126
Convolution  dec_ub3_rn2_conv2  1 1 b126 b127 0=128 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=147456
BinaryOp  dec_ub3_rn2_add  2 1 b127 b121 b128 0=0
GroupNorm  dec_norm_out  1 1 b128 b129 0=32 1=128 2=1e-06 3=1
Swish  dec_act_out  1 1 b129 b130
Convolution dec_conv_out 1 1 b130 out0 0=3 1=3 11=3 3=1 13=1 4=1 15=1 14=1 16=1 5=1 6=3456