# 扫出data path
import os
data_root_list = [
    # 'data/2_gmr_retarget_rl_0_120', 
    # 'data/2_gmr_retarget_rl_200_300', 
    # "data/2_gmr_retarget_rl_400_500"
    "/mnt/shenzhen2cephfs/capybarali/codes/neobot/data/motionmillion/2_gmr_retarget_full"
]
save_path = 'data/gmr_path.txt'

with open(save_path, 'w') as f:
    for data_root in data_root_list:
        for root, dirs, files in os.walk(data_root):
            for file in files:
                if file.endswith('.npz'):
                    f.write(os.path.join(root, file) + '\n')