FLOPS 썸네일형 리스트형 How to count the number of parameters of a NN and measure FLOPs required for the NN from fvcore.nn import FlopCountAnalysisdef count_parameters(model): return sum(p.numel() for p in model.parameters() if p.requires_grad) ResNet = ConvNet(use_pretrained=True, feature_extract=False, resent_model=saved_args.resnet_model)N_param = count_parameters(ResNet) / 1e6input_tensor = torch.randn(1, 3, 320, 640)flops = FlopCountAnalysis(ResNet, input_tensor.. 더보기 이전 1 다음