From eaffbbfffc40799e02c38a678b7e9d4344be4147 Mon Sep 17 00:00:00 2001 From: Larry Yan Date: Thu, 11 Jul 2019 19:56:32 +0800 Subject: [PATCH] fix(encoder): register tf inception in __init__ --- gnes/encoder/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gnes/encoder/__init__.py b/gnes/encoder/__init__.py index 75f620b2..c215c01a 100644 --- 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)