(cfg)
| 50 | |
| 51 | |
| 52 | def do_activation(cfg): |
| 53 | data_loader = build_detection_test_loader(cfg, cfg.DATASETS.TEST[0]) |
| 54 | model = build_model(cfg) |
| 55 | DetectionCheckpointer(model).load(cfg.MODEL.WEIGHTS) |
| 56 | model.eval() |
| 57 | |
| 58 | counts = Counter() |
| 59 | total_activations = [] |
| 60 | for idx, data in zip(tqdm.trange(args.num_inputs), data_loader): # noqa |
| 61 | count = activation_count_operators(model, data) |
| 62 | counts += count |
| 63 | total_activations.append(sum(count.values())) |
| 64 | logger.info( |
| 65 | "(Million) Activations for Each Type of Operators:\n" |
| 66 | + str([(k, v / idx) for k, v in counts.items()]) |
| 67 | ) |
| 68 | logger.info( |
| 69 | "Total (Million) Activations: {}±{}".format( |
| 70 | np.mean(total_activations), np.std(total_activations) |
| 71 | ) |
| 72 | ) |
| 73 | |
| 74 | |
| 75 | def do_parameter(cfg): |
nothing calls this directly
no test coverage detected