This commit is contained in:
语帆 2024-02-22 16:24:17 +08:00
parent 8cfee2db5c
commit cfc18a9047

View File

@ -164,9 +164,12 @@ class AutoModel:
pdb.set_trace()
# build frontend
frontend = kwargs.get("frontend", None)
pdb.set_trace()
if frontend is not None:
pdb.set_trace()
frontend_class = tables.frontend_classes.get(frontend)
frontend = frontend_class(**kwargs["frontend_conf"])
pdb.set_trace()
kwargs["frontend"] = frontend
kwargs["input_size"] = frontend.output_size()
pdb.set_trace()