(model)
| 80 | |
| 81 | |
| 82 | def count_parameters_in_MB(model): |
| 83 | return np.sum(np.prod(v.size()) for name, v in model.named_parameters() if "auxiliary" not in name)/1e6 |
| 84 | |
| 85 | |
| 86 | def save_checkpoint(state, is_best, save): |
nothing calls this directly
no outgoing calls
no test coverage detected