mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
fix format
This commit is contained in:
parent
01a85c5335
commit
298230177a
@ -25,25 +25,24 @@ import argparse
|
|||||||
import lhotse
|
import lhotse
|
||||||
from lhotse import load_manifest
|
from lhotse import load_manifest
|
||||||
|
|
||||||
|
|
||||||
def get_parser():
|
def get_parser():
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
||||||
)
|
)
|
||||||
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--input-manifest",
|
"--input-manifest", type=str, default="data/fbank/cuts_audioset_full.jsonl.gz"
|
||||||
type=str,
|
|
||||||
default="data/fbank/cuts_audioset_full.jsonl.gz"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--output",
|
"--output",
|
||||||
type=str,
|
type=str,
|
||||||
required=True,
|
required=True,
|
||||||
|
|
||||||
)
|
)
|
||||||
return parser
|
return parser
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
# Reference: https://github.com/YuanGongND/ast/blob/master/egs/audioset/gen_weight_file.py
|
# Reference: https://github.com/YuanGongND/ast/blob/master/egs/audioset/gen_weight_file.py
|
||||||
parser = get_parser()
|
parser = get_parser()
|
||||||
@ -69,5 +68,6 @@ def main():
|
|||||||
weight += 1000 / (label_count[label] + 0.01)
|
weight += 1000 / (label_count[label] + 0.01)
|
||||||
f.write(f"{c.id} {weight}\n")
|
f.write(f"{c.id} {weight}\n")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user