jackroos
/ compute_flops.py
Created
June 29, 2020 11:34
— forked from fmassa/compute_flops.py
Utility functions used to compute flops in DETR.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # this is the main entrypoint | |
| # as we describe in the paper, we compute the flops over the first 100 images | |
| # on COCO val2017, and report the average result | |
| import torch | |
| import time | |
| import torchvision | |
| import numpy as np | |
| import tqdm |