Update registry with Wave 1 solvers
Browse files
neurogolf_solver/solvers/solver_registry.py
CHANGED
|
@@ -12,6 +12,8 @@ from .tiling import (s_tile, s_upscale, s_kronecker, s_nonuniform_scale, s_diago
|
|
| 12 |
from .gravity import s_gravity_unrolled
|
| 13 |
from .edge import s_edge_detect
|
| 14 |
from .mode import s_mode_fill
|
|
|
|
|
|
|
| 15 |
from .conv import solve_conv_fixed, solve_conv_variable, solve_conv_diffshape, solve_conv_var_diff
|
| 16 |
from ..data_loader import get_exs, fixed_shapes
|
| 17 |
from ..validators import validate
|
|
@@ -43,6 +45,12 @@ ANALYTICAL_SOLVERS = [
|
|
| 43 |
('gravity_unrolled', s_gravity_unrolled),
|
| 44 |
('edge_detect', s_edge_detect),
|
| 45 |
('mode_fill', s_mode_fill),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
]
|
| 47 |
|
| 48 |
|
|
|
|
| 12 |
from .gravity import s_gravity_unrolled
|
| 13 |
from .edge import s_edge_detect
|
| 14 |
from .mode import s_mode_fill
|
| 15 |
+
from .wave1 import (s_downsample_stride, s_symmetry_complete, s_extract_inner,
|
| 16 |
+
s_add_border, s_sparse_fill, s_channel_filter)
|
| 17 |
from .conv import solve_conv_fixed, solve_conv_variable, solve_conv_diffshape, solve_conv_var_diff
|
| 18 |
from ..data_loader import get_exs, fixed_shapes
|
| 19 |
from ..validators import validate
|
|
|
|
| 45 |
('gravity_unrolled', s_gravity_unrolled),
|
| 46 |
('edge_detect', s_edge_detect),
|
| 47 |
('mode_fill', s_mode_fill),
|
| 48 |
+
('downsample_stride', s_downsample_stride),
|
| 49 |
+
('symmetry_complete', s_symmetry_complete),
|
| 50 |
+
('extract_inner', s_extract_inner),
|
| 51 |
+
('add_border', s_add_border),
|
| 52 |
+
('sparse_fill', s_sparse_fill),
|
| 53 |
+
('channel_filter', s_channel_filter),
|
| 54 |
]
|
| 55 |
|
| 56 |
|