游雁
2024-02-19 ff4306346eae4021c711df3fe23979e82e06e751
funasr/bin/train.py
@@ -80,8 +80,7 @@
        kwargs["frontend"] = frontend
        kwargs["input_size"] = frontend.output_size()
    
    # import pdb;
    # pdb.set_trace()
    # build model
    model_class = tables.model_classes.get(kwargs["model"])
    model = model_class(**kwargs, **kwargs["model_conf"], vocab_size=len(tokenizer.token_list))