Fixes incorrect batch_size calculation

experimenti · Feb 1, 2018 · 291bd2b · 291bd2b
1 parent 7b4d259
commit 291bd2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tune_decoder.py b/tune_decoder.py
@@ -92,7 +92,7 @@ def decode_dataset(logits, test_dataset, batch_size, lm_alpha, lm_beta, mesh_x,
  normalize=True)
 
  logits = np.load(args.logits)
- batch_size = logits[0][0].shape[1]
+ batch_size = logits[0][0].shape[0]
 
  results = []