(model_name_lst, data_name_lst)
| 59 | |
| 60 | |
| 61 | def gather_res(model_name_lst, data_name_lst): |
| 62 | res = [] |
| 63 | for model_name in model_name_lst: |
| 64 | for data_name in data_name_lst: |
| 65 | with open('/home/jindwang/mine/clipood/log/{}_{}.txt'.format(model_name, data_name), 'r') as f: |
| 66 | lines = f.readlines() |
| 67 | line = lines[-1].strip().split(' ')[-1] |
| 68 | line = '{},{},{}'.format(model_name, data_name, line) |
| 69 | res.append(line) |
| 70 | return res |
| 71 | |
| 72 | |
| 73 | def convert_models_to_fp32(model): |