rogermt commited on
Commit
1fae8b5
·
verified ·
1 Parent(s): 48b10b6

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