refactor(base): better batch_size control

gnes-ai · Aug 2, 2019 · c430ef6 · c430ef6
1 parent 8b72fd9
commit c430ef6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/gnes/encoder/video/incep_mixture.py b/gnes/encoder/video/incep_mixture.py
@@ -102,7 +102,6 @@ def post_init(self):
             saver.restore(self.sess2, self.model_dir_mixture)
 
     def encode(self, data: List['np.ndarray'], *args, **kwargs) -> np.ndarray:
-        ret = []
         v_len = [len(v) for v in data]
         pos_start = [0] + [sum(v_len[:i + 1]) for i in range(len(v_len) - 1)]
         pos_end = [sum(v_len[:i + 1]) for i in range(len(v_len))]

diff --git a/tests/yaml/router-batch.yml b/tests/yaml/router-batch.yml
@@ -1,3 +1,3 @@
 !DocBatchRouter
-parameter:
+gnes_config:
   batch_size: 2