提交 9de03438 编写于 作者: H Hui Zhang

fix augment

上级 9dace625
......@@ -97,7 +97,7 @@ class AugmentationPipeline():
ValueError: If the augmentation json config is in incorrect format".
"""
SPEC_TYPES = ('specaug')
SPEC_TYPES = {'specaug'}
def __init__(self, augmentation_config: str, random_seed: int=0):
self._rng = np.random.RandomState(random_seed)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册