Skip to content

Instantly share code, notes, and snippets.

View PistonY's full-sized avatar
🎈
CS PhD student at Illinois Tech. Focus on computer vision.

Devin Yang PistonY

🎈
CS PhD student at Illinois Tech. Focus on computer vision.
  • XiaoMi
  • China BeiJing
View GitHub Profile
@PistonY
PistonY / flops.py
Created May 23, 2019 09:15
Flops for Gluon
# -*- coding: utf-8 -*-
# Author: pistonyang@gmail.com
from collections import OrderedDict
from mxnet import ndarray
from mxnet.gluon.nn import HybridBlock
def summary(block, *inputs):
"""Print the summary of the model's output and parameters.