(model, verbose=False)
| 63 | |
| 64 | |
| 65 | def count_params(model, verbose=False): |
| 66 | total_params = sum(p.numel() for p in model.parameters()) |
| 67 | if verbose: |
| 68 | print(f"{model.__class__.__name__} has {total_params*1.e-6:.2f} M params.") |
| 69 | return total_params |
| 70 | |
| 71 | |
| 72 | def instantiate_from_config(config): |