(params)
| 23 | print(f"Seed set to: {seed} (type: {type(seed)})") |
| 24 | |
| 25 | def params2cpu(params): |
| 26 | res = {} |
| 27 | for k, v in params.items(): |
| 28 | if isinstance(v, torch.Tensor): |
| 29 | res[k] = v.detach().cpu().contiguous().numpy() |
| 30 | else: |
| 31 | res[k] = v |
| 32 | return res |
| 33 | |
| 34 | def save_params(output_params, output_dir): |
| 35 | # Convert to CPU Numpy Arrays |
no outgoing calls
no test coverage detected