Commit 02a5d258 authored by Yuxin Wu's avatar Yuxin Wu

fix build

parent 300d840a
...@@ -3,13 +3,12 @@ ...@@ -3,13 +3,12 @@
# File: dump-model-params.py # File: dump-model-params.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com> # Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np
import argparse import argparse
import tensorflow as tf import tensorflow as tf
import imp import imp
from tensorpack import TowerContext, logger from tensorpack import TowerContext, logger
from tensorpack.tfutils import sessinit, varmanip, get_model_loader from tensorpack.tfutils import varmanip, get_model_loader
from tensorpack.graph_builder.input_source import PlaceholderInput from tensorpack.graph_builder.input_source import PlaceholderInput
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
......
...@@ -183,10 +183,12 @@ def load_chkpt_vars(model_path): ...@@ -183,10 +183,12 @@ def load_chkpt_vars(model_path):
result[n] = reader.get_tensor(n) result[n] = reader.get_tensor(n)
return result return result
def dump_chkpt_vars(model_path): def dump_chkpt_vars(model_path):
logger.warn("dump_chkpt_vars was renamed to load_chkpt_vars!") logger.warn("dump_chkpt_vars was renamed to load_chkpt_vars!")
return load_chkpt_vars(model_path) return load_chkpt_vars(model_path)
def is_training_name(name): def is_training_name(name):
""" """
Guess if a name belongs to a training-only variables. Guess if a name belongs to a training-only variables.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment