“3d23804ec799ec071670341731466dc2d91a88eb”上不存在“doc/tutorials/image_classification/index_en.html”
提交 d2fab323 编写于 作者: J Junkun

fix bugs

上级 e867f3bb
......@@ -102,10 +102,10 @@ def read_manifest(
manifest = []
with jsonlines.open(manifest_path, 'r') as reader:
for json_data in reader:
feat_len = json_data["feat_shape"][
0] if 'feat_shape' in json_data else 1.0
token_len = json_data["token_shape"][
0] if 'token_shape' in json_data else 1.0
feat_len = json_data["input"][0]["shape"][
0] if 'shape' in json_data["input"][0] else 1.0
token_len = json_data["output"][0]["shape"][
0] if 'shape' in json_data["output"][0] else 1.0
conditions = [
feat_len >= min_input_len,
feat_len <= max_input_len,
......
......@@ -51,7 +51,7 @@ def _batch_shuffle(indices, batch_size, epoch, clipped=False):
"""
rng = np.random.RandomState(epoch)
shift_len = rng.randint(0, batch_size - 1)
batch_indices = list(zip(* [iter(indices[shift_len:])] * batch_size))
batch_indices = list(zip(*[iter(indices[shift_len:])] * batch_size))
rng.shuffle(batch_indices)
batch_indices = [item for batch in batch_indices for item in batch]
assert clipped is False
......
......@@ -94,6 +94,9 @@ class Checkpoint():
"""
configs = {}
if len(checkpoint_path) == 0 or checkpoint_path == "None":
checkpoint_path = None
if checkpoint_path is not None:
pass
elif checkpoint_dir is not None and record_file is not None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册