(model, verbose=False)
| 69 | |
| 70 | |
| 71 | def count_params(model, verbose=False): |
| 72 | total_params = sum(p.numel() for p in model.parameters()) |
| 73 | if verbose: |
| 74 | print(f"{model.__class__.__name__} has {total_params * 1.e-6:.2f} M params.") |
| 75 | return total_params |
| 76 | |
| 77 | |
| 78 | def instantiate_from_config(config): |
no outgoing calls
no test coverage detected