Skip to content
Permalink
Browse files

[DoReFa] remove memoization & warning, they are not needed after havi…

…ng custom_gradient
  • Loading branch information...
ppwwyyxx committed Apr 10, 2019
1 parent f831a46 commit 8dd254be676d89000a7f4f6fd5cebe78b3499789
Showing with 0 additions and 4 deletions.
  1. +0 −4 examples/DoReFa-Net/dorefa.py
@@ -4,14 +4,10 @@

import tensorflow as tf

from tensorpack.utils.argtools import graph_memoized


@graph_memoized
def get_dorefa(bitW, bitA, bitG):
"""
Return the three quantization functions fw, fa, fg, for weights, activations and gradients respectively
It's unsafe to call this function multiple times with different parameters
"""
def quantize(x, k):
n = float(2 ** k - 1)

0 comments on commit 8dd254b

Please sign in to comment.
You can’t perform that action at this time.