(model, verbose=False)
| 2 | |
| 3 | |
| 4 | def count_params(model, verbose=False): |
| 5 | total_params = sum(p.numel() for p in model.parameters()) |
| 6 | if verbose: |
| 7 | print(f"{model.__class__.__name__} has {total_params*1.e-6:.2f} M params.") |
| 8 | return total_params |
| 9 | |
| 10 | |
| 11 | def instantiate_from_config(config): |
nothing calls this directly
no outgoing calls
no test coverage detected