espnet2.sds.end_to_end.mini_omni.litgpt.utils.flops_per_param

source

espnet2.sds.end_to_end.mini_omni.litgpt.utils.flops_per_param(max_seq_length: int, n_layer: int, n_embd: int, n_params: int) → int