Files
llm_trainer/scripts/cal_flops.py