Files
LEANN/research/micro/save_small_model.py