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()