inference code

[godfunc123]

Is there a code for the inference ？especially for the experiment analyzed the average inference time and GFLOPs, Thank you

[FengheTan9]

Thanks for using our code.
You can use the following code for analyzing:
from fvcore.nn import FlopCountAnalysis, parameter_count_table
flops = FlopCountAnalysis(model, input)
or
from thop import profile
flops, params = profile(model, inputs=(input, ))

[godfunc123]

Thank you