(samples)
| 162 | |
| 163 | |
| 164 | def unpack_dict_to_list(samples): |
| 165 | samples_ = [] |
| 166 | bsz = samples.get('outputs').size(0) |
| 167 | for i in range(bsz): |
| 168 | res = {} |
| 169 | for k, v in samples.items(): |
| 170 | try: |
| 171 | res[k] = v[i] |
| 172 | except: |
| 173 | pass |
| 174 | samples_.append(res) |
| 175 | return samples_ |
| 176 | |
| 177 | |
| 178 | def load_ckpt(cur_model, ckpt_base_dir, prefix_in_ckpt='model', force=True, strict=True): |