def METHOD_NAME(
    cls,
    additional_dir: Workspace,
    workspace: Workspace,
    task: Task,
):
    setup_problem_cfg = workspace.get_cfg(cls.setup_problem)
    setup_problem_cfg["upstream"].pop("_cls")
    setup_problem_cfg["upstream"].pop("input_size")
    apc_config = dict(
        model=dict(
            paras=setup_problem_cfg["upstream"],
        ),
        data=dict(
            audio=_audio_config,
        ),
    )
    all_states = dict(
        config=apc_config,
        model=task.upstream.state_dict(),
        Upstream_Config=apc_config,
    )
    torch.save(
        all_states, str(additional_dir.parent.resolve()) + "/all_states.ckpt"
    )