国外玩家整理的一份SAEHD模型的参数表格,大家可以参考下,根据自己的显卡设置对应的模型参数
注释:架构即模型架构,英文叫archi;分辨率即模型分辨率,英文叫resolution;AE/E/D/MD即ae_dims/e_dims/d_dims/d_mask_dims;批量大小即batch_size;延时即Iter;GPU训练即models_opt_on_gpu;AB优化器即adabelief
显卡
|
显存
|
CPU
|
内存
|
架构
|
分支
|
分辨率
|
AE Dim
|
E Dim
|
D Dim
|
M Dim
|
批量大小
|
延时
|
GPU训练
|
AB优化器
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UD
|
320
|
320
|
96
|
96
|
32
|
7
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UD
|
320
|
360
|
90
|
90
|
22
|
7
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UDT
|
320
|
320
|
88
|
88
|
22
|
7
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UD
|
320
|
288
|
72
|
72
|
22
|
7
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UDT
|
320
|
320
|
88
|
88
|
22
|
7
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UD
|
320
|
288
|
80
|
80
|
22
|
9
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UDT
|
256
|
300
|
80
|
64
|
22
|
17
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UDT
|
288
|
300
|
80
|
80
|
22
|
10
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
DF
|
UDT
|
320
|
360
|
90
|
90
|
22
|
5
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UD
|
256
|
256
|
64
|
64
|
22
|
18
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UD
|
256
|
256
|
80
|
80
|
22
|
13
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UD
|
320
|
256
|
64
|
64
|
22
|
10
|
1350
|
True
|
True
|
RTX 3060
|
12
|
i5-8400
|
32
|
LIAE
|
UDT
|
320
|
256
|
72
|
72
|
32
|
9
|
1350
|
True
|
True
|
GTX Titan X
|
12
|
i7-2600K
|
24
|
LIAE
|
UDT
|
224
|
256
|
64
|
64
|
16
|
6
|
1035
|
True
|
True
|
RTX 3050
|
8
|
i5-4670
|
8
|
DF
|
UD
|
320
|
256
|
64
|
64
|
22
|
4
|
680
|
True
|
True
|
RTX 3090
|
24
|
R9-3900X
|
32
|
LIAE
|
UDT
|
224
|
512
|
64
|
64
|
16
|
48
|
998
|
True
|
True
|
RTX 3090
|
24
|
R9-3900X
|
32
|
LIAE
|
UDT
|
288
|
352
|
128
|
128
|
16
|
16
|
1320
|
True
|
True
|
RTX 3090
|
24
|
R9-3900X
|
32
|
LIAE
|
UD
|
384
|
384
|
116
|
116
|
16
|
6
|
942
|
True
|
False
|
RTX 3090
|
24
|
R9-3900X
|
32
|
DF
|
UD
|
384
|
512
|
112
|
112
|
16
|
8
|
1013
|
True
|
True
|
RTX 3090
|
24
|
R3-3900X
|
32
|
DF
|
UD
|
320
|
512
|
112
|
112
|
16
|
16
|
1074
|
True
|
True
|
RTX 3060
|
12
|
R5-5600X
|
32
|
DF
|
UD
|
384
|
256
|
64
|
64
|
22
|
8
|
1140
|
True
|
True
|
RTX 3060
|
12
|
R5-3600
|
32
|
DF
|
UD
|
384
|
256
|
64
|
64
|
22
|
8
|
1130
|
True
|
True
|
RTX 2080
|
8
|
i7-8700K
|
32
|
LIAE
|
UD
|
256
|
256
|
64
|
64
|
22
|
8
|
500
|
False
|
True
|
Tesla V100
|
16
|
Colab
|
25
|
DF
|
UD
|
384
|
352
|
88
|
88
|
16
|
8
|
1000
|
True
|
True
|
Tesla V100
|
16
|
Colab
|
25
|
DF
|
UD
|
320
|
416
|
104
|
104
|
16
|
8
|
850
|
True
|
True
|
Tesla V100
|
16
|
Colab
|
25
|
DF
|
UD
|
384
|
320
|
80
|
80
|
22
|
8
|
900
|
True
|
True
|
Tesla V100
|
16
|
Colab
|
25
|
DF
|
UD
|
448
|
256
|
64
|
64
|
22
|
8
|
950
|
True
|
True
|
GTX 1060
|
6
|
i5-4690K
|
16
|
LIAE
|
UD
|
320
|
264
|
72
|
72
|
24
|
5
|
6500
|
False
|
True
|
GTX 1080 Ti
|
11
|
i7-4770K
|
16
|
DF
|
UD
|
320
|
320
|
72
|
72
|
16
|
8
|
700
|
True
|
True
|
RTX 3090
|
24
|
i7-5820K
|
48
|
DF
|
UD
|
416
|
416
|
104
|
104
|
26
|
8
|
1170
|
True
|
True
|
RTX 3090
|
24
|
i7-9700K
|
16
|
DF
|
UD
|
256
|
256
|
64
|
64
|
22
|
16
|
581
|
True
|
False
|
GTX 1070
|
8
|
i7-7700HQ
|
16
|
LIAE
|
UD
|
224
|
288
|
96
|
96
|
16
|
4
|
800
|
True
|
False
|
GTX 1080 Ti
|
11
|
W3680
|
12
|
DF
|
UD
|
288
|
256
|
80
|
80
|
20
|
12
|
1205
|
True
|
False
|
GTX 1060
|
6
|
i7-6700HQ
|
16
|
DF
|
UD
|
256
|
320
|
96
|
96
|
22
|
4
|
1800
|
True
|
False
|
Tesla P100
|
16
|
Colab
|
16
|
DF
|
UD
|
256
|
320
|
96
|
96
|
22
|
4
|
460
|
True
|
False
|
GTX 1080 Ti
|
11
|
W3680
|
12
|
DF
|
UD
|
288
|
256
|
80
|
80
|
20
|
4
|
484
|
True
|
False
|
GTX 1080 Ti
|
11
|
W3680
|
12
|
DF
|
UD
|
288
|
256
|
80
|
80
|
20
|
6
|
719
|
True
|
False
|
GTX 1080 Ti
|
11
|
W3680
|
12
|
DF
|
UD
|
288
|
256
|
80
|
80
|
20
|
8
|
850
|
True
|
False
|
GTX 1080 Ti
|
11
|
W3680
|
12
|
DF
|
UD
|
288
|
256
|
80
|
80
|
20
|
8
|
862
|
True
|
False
|
RTX 2080
|
8
|
R7-3800X
|
16
|
DF
|
UD
|
320
|
256
|
64
|
64
|
22
|
5
|
478
|
True
|
False
|
RTX 2080 Ti
|
11
|
i7-9700K
|
16
|
DF
|
UD
|
256
|
256
|
64
|
64
|
22
|
20
|
800
|
True
|
False
|
RTX 2060 S
|
8
|
R5-3500
|
16
|
DF
|
UD
|
256
|
256
|
64
|
64
|
22
|
14
|
800
|
True
|
False
|
GTX 1050 Ti
|
4
|
E5-1620
|
16
|
LIAE
|
–
|
128
|
128
|
80
|
48
|
16
|
4
|
520
|
True
|
False
|
Tesla P100
|
16
|
Colab
|
16
|
DF
|
–
|
192
|
768
|
80
|
80
|
22
|
8
|
1000
|
True
|
False
|
Titan RTX
|
24
|
RT-3970X
|
128
|
DF
|
–
|
400
|
256
|
64
|
64
|
22
|
6
|
1350
|
True
|
False
|
Titan RTX (2)
|
48
|
RT-3970X
|
128
|
DF
|
–
|
512
|
256
|
64
|
64
|
22
|
6
|
1700
|
True
|
False
|
Titan RTX (2)
|
48
|
RT-3970X
|
128
|
DF
|
–
|
512
|
256
|
64
|
64
|
22
|
8
|
2100
|
False
|
False
|
Titan RTX (2)
|
48
|
RT-3970X
|
128
|
DF
|
–
|
400
|
256
|
64
|
64
|
22
|
12
|
2200
|
True
|
False
|
GTX 1050 Ti
|
4
|
i5-3570K
|
12
|
DF
|
–
|
128
|
192
|
48
|
48
|
36
|
2
|
645
|
True
|
False
|
GTX 1050 Ti
|
4
|
i5-3570K
|
12
|
DF
|
–
|
128
|
192
|
48
|
48
|
36
|
2
|
645
|
False
|
False
|
GTX 970
|
4
|
i7-2600K
|
12
|
DF
|
–
|
96
|
256
|
64
|
64
|
22
|
4
|
700
|
True
|
False
|
GTX 970
|
4
|
i7-2600K
|
12
|
DF
|
–
|
96
|
256
|
64
|
64
|
22
|
4
|
700
|
True
|
False
|
Quadro M2200
|
4
|
E3-1535M v6
|
32
|
DF
|
–
|
128
|
512
|
64
|
48
|
16
|
4
|
921
|
True
|
False
|
RTX 2070
|
8
|
R7-3800X
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
32
|
8
|
1100
|
False
|
False
|
RTX 2080
|
8
|
i7-8700
|
16
|
DF
|
–
|
224
|
512
|
80
|
80
|
22
|
2
|
406
|
True
|
False
|
RTX 2080
|
8
|
i7-8700
|
16
|
DF
|
–
|
192
|
512
|
64
|
64
|
22
|
7
|
570
|
True
|
False
|
RTX 2080
|
8
|
i7-8700
|
16
|
DF
|
–
|
192
|
512
|
80
|
80
|
26
|
3
|
570
|
True
|
False
|
RTX 2080
|
8
|
i7-8700
|
16
|
DF
|
–
|
224
|
512
|
64
|
64
|
22
|
5
|
580
|
True
|
False
|
GTX 1080
|
8
|
i7-8700K
|
32
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
8
|
780
|
True
|
False
|
GTX 1650
|
4
|
i5-9300H
|
16
|
DF
|
–
|
128
|
256
|
64
|
64
|
22
|
6
|
780
|
True
|
False
|
GTX 1650
|
4
|
i5-9300H
|
16
|
DF
|
–
|
128
|
256
|
64
|
64
|
22
|
8
|
960
|
False
|
False
|
GTX 1070 Ti
|
8
|
R5-3600
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
10
|
1030
|
False
|
False
|
Tesla P100
|
16
|
Colab
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
18
|
1200
|
True
|
False
|
GTX 1060
|
6
|
i5-4670K
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
7
|
1200
|
False
|
False
|
GTX 1060
|
6
|
i5-4590
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
6
|
1400
|
False
|
False
|
RTX 2070
|
8
|
i7-8700
|
16
|
DF
|
–
|
144
|
256
|
64
|
64
|
22
|
8
|
400
|
True
|
False
|
Titan RTX
|
24
|
E5-1650
|
64
|
DF
|
–
|
256
|
256
|
64
|
64
|
22
|
16
|
4100
|
True
|
False
|
Titan RTX
|
24
|
E5-1650
|
64
|
DF
|
–
|
224
|
256
|
64
|
64
|
22
|
20
|
4600
|
True
|
False
|
RTX 2070 S
|
8
|
R5-3600
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
5
|
600
|
False
|
False
|
GTX 1660 Ti
|
6
|
i7-9700
|
16
|
DF
|
–
|
192
|
256
|
64
|
48
|
22
|
8
|
676
|
True
|
False
|
GTX 1060
|
6
|
i5-4670K
|
16
|
DF
|
–
|
128
|
450
|
64
|
64
|
22
|
10
|
750
|
True
|
False
|
Tesla P100
|
16
|
Colab
|
16
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
12
|
800
|
True
|
False
|
RTX 2060
|
6
|
i5-2500K
|
8
|
DF
|
–
|
160
|
256
|
64
|
64
|
22
|
6
|
2500
|
False
|
False
|
RTX 2060
|
6
|
i5-2500K
|
8
|
DF
|
–
|
128
|
256
|
64
|
64
|
22
|
14
|
600
|
True
|
False
|
RTX 2060
|
6
|
R5-2600
|
16
|
LIAE
|
–
|
256
|
256
|
64
|
64
|
22
|
2
|
1700
|
False
|
False
|
RTX 2080 Ti
|
11
|
i9-9900K
|
32
|
LIAE
|
U
|
256
|
256
|
64
|
64
|
22
|
6
|
700
|
True
|
False
|
RTX 2080 Ti (2)
|
22
|
R7-2700
|
32
|
LIAE
|
–
|
192
|
256
|
64
|
64
|
22
|
20
|
1230
|
True
|
False
|
GTX 750 Ti
|
2
|
i5-4690K
|
32
|
LIAE
|
–
|
112
|
256
|
64
|
64
|
22
|
4
|
1450
|
False
|
False
|
GTX 1080 Ti
|
11
|
i5-4590
|
16
|
LIAE
|
–
|
192
|
256
|
64
|
64
|
22
|
8
|
670
|
True
|
False
|
GTX 1080 Ti
|
11
|
i5-4590
|
16
|
LIAE
|
–
|
192
|
256
|
64
|
64
|
22
|
12
|
900
|
True
|
False
|
GTX 1080 Ti
|
11
|
i7-4770K
|
16
|
DF
|
UD
|
352
|
288
|
72
|
72
|
16
|
10
|
1200
|
True
|
False
|
RTX 2070 S
|
8
|
i7-9700K
|
16
|
DF
|
U
|
192
|
256
|
64
|
64
|
22
|
9
|
670
|
True
|
False
|
RTX 2080ti
|
11
|
i9-9900K
|
32
|
DF
|
–
|
192
|
256
|
64
|
64
|
22
|
12
|
536
|
True
|
False
|
RTX 2060
|
6
|
RT-1950x
|
32
|
DF
|
–
|
128
|
128
|
32
|
32
|
22
|
8
|
215
|
False
|
False
|
GTX 1080 Ti
|
11
|
R7-3700X
|
16
|
DF
|
–
|
192
|
256
|
64
|
48
|
18
|
4
|
510
|
True
|
False
|
GTX 1650
|
4
|
i7-9750H
|
16
|
DF
|
–
|
128
|
256
|
64
|
64
|
22
|
4
|
792
|
False
|
False
|
RTX 2060
|
6
|
RT-1950X
|
32
|
DF
|
–
|
128
|
128
|
32
|
32
|
22
|
1
|
85
|
False
|
False
|