fix(encoder): register tf inception in __init__

gnes-ai · Jul 11, 2019 · eaffbbf · eaffbbf
1 parent d0099b7
commit eaffbbf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gnes/encoder/__init__.py b/gnes/encoder/__init__.py
@@ -39,6 +39,7 @@
     'PipelineEncoder': 'base',
     'HashEncoder': 'numeric.hash',
     'BasePytorchEncoder': 'image.base',
+    'TFInceptionEncoder': 'image.inception',
 }
 
 register_all_class(_cls2file_map)