Skip to content

Instantly share code, notes, and snippets.

View iwiwi's full-sized avatar
Working from home

Takuya Akiba iwiwi

Working from home
View GitHub Profile
t-abe /
Created April 28, 2017 09:14
Hook implementation to calculate roughly estimated computational cost of a given network for Chainer
from __future__ import print_function
import sys
import chainer
from chainer.utils import conv
class ComputationalCostHook(chainer.function.FunctionHook):
name = 'ComputationalCostHook'