File tree Expand file tree Collapse file tree 1 file changed +34
-0
lines changed
DeepCrazyhouse/src/domain/neural_net/architectures/pytorch Expand file tree Collapse file tree 1 file changed +34
-0
lines changed Original file line number Diff line number Diff line change @@ -65,6 +65,40 @@ def get_alpha_vile_model(args, model_size='normal'):
65
65
kernels [33 ] = 5
66
66
kernels [34 ] = 5
67
67
kernels [35 ] = 5
68
+ elif model_size == 'normal' :
69
+ kernels [0 ] = 5
70
+ kernels [4 ] = 5
71
+ kernels [5 ] = 5
72
+ kernels [7 ] = 5
73
+ kernels [10 ] = 5
74
+ kernels [12 ] = 5
75
+ kernels [15 ] = 5
76
+ kernels [16 ] = 5
77
+ kernels [20 ] = 5
78
+ kernels [21 ] = 5
79
+ kernels [22 ] = 5
80
+ kernels [23 ] = 5
81
+ kernels [24 ] = 5
82
+ kernels [25 ] = 5
83
+ elif model_size == 'small' :
84
+ kernels [0 ] = 5
85
+ kernels [4 ] = 5
86
+ kernels [5 ] = 5
87
+ kernels [6 ] = 5
88
+ kernels [10 ] = 5
89
+ kernels [11 ] = 5
90
+ kernels [13 ] = 5
91
+ kernels [15 ] = 5
92
+ kernels [17 ] = 5
93
+ kernels [20 ] = 5
94
+ elif model_size == 'tiny' :
95
+ kernels [1 ] = 5
96
+ kernels [2 ] = 5
97
+ kernels [6 ] = 5
98
+ kernels [7 ] = 5
99
+ kernels [8 ] = 5
100
+ kernels [9 ] = 5
101
+ kernels [10 ] = 5
68
102
else :
69
103
for idx in range (end_idx ):
70
104
kernels [idx ] = 5
You can’t perform that action at this time.
0 commit comments