Skip to content
This repository has been archived by the owner on Feb 22, 2020. It is now read-only.

Commit

Permalink
Browse files Browse the repository at this point in the history
refactor(base): better batch_size control
  • Loading branch information
hanhxiao committed Aug 2, 2019
1 parent 8b72fd9 commit c430ef6
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
1 change: 0 additions & 1 deletion gnes/encoder/video/incep_mixture.py
Expand Up @@ -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))]
Expand Down
2 changes: 1 addition & 1 deletion tests/yaml/router-batch.yml
@@ -1,3 +1,3 @@
!DocBatchRouter
parameter:
gnes_config:
batch_size: 2

0 comments on commit c430ef6

Please sign in to comment.