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

fix build

parent 300d840a
......@@ -3,13 +3,12 @@
# File: dump-model-params.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np
import argparse
import tensorflow as tf
import imp
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
parser = argparse.ArgumentParser()
......
......@@ -183,10 +183,12 @@ def load_chkpt_vars(model_path):
result[n] = reader.get_tensor(n)
return result
def dump_chkpt_vars(model_path):
logger.warn("dump_chkpt_vars was renamed to load_chkpt_vars!")
return load_chkpt_vars(model_path)
def is_training_name(name):
"""
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