(model, verbose=False)
| 159 | |
| 160 | |
| 161 | def count_params(model, verbose=False): |
| 162 | total_params = sum(p.numel() for p in model.parameters()) |
| 163 | if verbose: |
| 164 | print(f"{model.__class__.__name__} has {total_params * 1.e-6:.2f} M params.") |
| 165 | return total_params |
| 166 | |
| 167 | |
| 168 | def instantiate_from_config(config): |