(params)
| 21 | |
| 22 | |
| 23 | def _to_tensor(params): |
| 24 | transformed = {} |
| 25 | |
| 26 | for k, v in params.items(): |
| 27 | if not isinstance(v, Tensor): |
| 28 | transformed[k] = tensor(v) |
| 29 | else: |
| 30 | transformed[k] = v |
| 31 | |
| 32 | return transformed |
| 33 | |
| 34 | |
| 35 | def save_param_dict(params): |
no test coverage detected
searching dependent graphs…