Skip to content

Commit 09b5b5a

Browse files
committed
Fix kernel 5x5 for AlphaVile (normal, small, tiny)
1 parent 81aa7fd commit 09b5b5a

File tree

1 file changed

+34
-0
lines changed
  • DeepCrazyhouse/src/domain/neural_net/architectures/pytorch

1 file changed

+34
-0
lines changed

DeepCrazyhouse/src/domain/neural_net/architectures/pytorch/alpha_vile.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,40 @@ def get_alpha_vile_model(args, model_size='normal'):
6565
kernels[33] = 5
6666
kernels[34] = 5
6767
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
68102
else:
69103
for idx in range(end_idx):
70104
kernels[idx] = 5

0 commit comments

Comments
 (0)