Commit 5ad84461 authored by Yuxin Wu's avatar Yuxin Wu

fix "dump-model-params.py"

parent 8a18bfe4
......@@ -7,9 +7,8 @@ import argparse
import tensorflow as tf
import imp
from tensorpack import TowerContext, logger
from tensorpack import TowerContext, logger, PlaceholderInput
from tensorpack.tfutils import varmanip, get_model_loader
from tensorpack.graph_builder.input_source import PlaceholderInput
parser = argparse.ArgumentParser()
parser.add_argument('--config', help='config file')
......
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