rtf benchmark

游雁

2023-03-17 c437307a8bfb8cc5935b7c2c00d12f2b6f4293ed

 funasr/datasets/large_datasets/utils/padding.py

@@ -6,9 +6,8 @@
def padding(data, float_pad_value=0.0, int_pad_value=-1):
    assert isinstance(data, list)
    assert "key" in data[0]
    assert "speech" in data[0]
    assert "text" in data[0]

    assert "speech" in data[0] or "text" in data[0]
    
    keys = [x["key"] for x in data]

    batch = {}