| 178 | |
| 179 | |
| 180 | def to_float32(preds): |
| 181 | if isinstance(preds, dict): |
| 182 | for k in preds: |
| 183 | if isinstance(preds[k], dict) or isinstance(preds[k], list): |
| 184 | preds[k] = to_float32(preds[k]) |
| 185 | elif isinstance(preds[k], paddle.Tensor): |
| 186 | preds[k] = preds[k].astype(paddle.float32) |
| 187 | elif isinstance(preds, list): |
| 188 | for k in range(len(preds)): |
| 189 | if isinstance(preds[k], dict): |
| 190 | preds[k] = to_float32(preds[k]) |
| 191 | elif isinstance(preds[k], list): |
| 192 | preds[k] = to_float32(preds[k]) |
| 193 | elif isinstance(preds[k], paddle.Tensor): |
| 194 | preds[k] = preds[k].astype(paddle.float32) |
| 195 | elif isinstance(preds, paddle.Tensor): |
| 196 | preds = preds.astype(paddle.float32) |
| 197 | return preds |
| 198 | |
| 199 | |
| 200 | def train( |