Slide 24
Slide 24 text
モデルの計算量を算出してくれる
Chainer Computational Cost
https://github.com/belltailjp/chainer_computational_cost
import chainer
import numpy as np
from chainer_computational_cost import
ComputationalCostHook
net = chainer.links.VGG16Layers()
x = np.random.random((1, 3, 224, 224)).astype(np.float32)
with chainer.no_backprop_mode(),
chainer.using_config('train', False):
with ComputationalCostHook(fma_1flop=True) as cch:
y = net(x)
cch.show_report(unit='G', mode='md')