def flop(model, weights, orig=15346630656): setting = getattr(cfgs, args.setting) if model == '': model = setting.model if weights == '': weights = setting.weights orig = setting.flop print('orig', orig) net = Net(model, model=weights, noTF=1) after = net.computation() print(after * 100 / orig)
def compute(pt='../resnet-cifar10-caffe/resnet-56/trainval.prototxt', model="../resnet-cifar10-caffe/resnet-56/snapshot/_iter_64000.caffemodel"): net = Net(pt, model=model, noTF=1) net.computation()