Commit a0619cd6 authored by Yuxin Wu's avatar Yuxin Wu

clean-up imports and docs

parent bf57d0af
tensorpack.RL package
=====================
.. automodule:: tensorpack.RL
:members:
:undoc-members:
:show-inheritance:
......@@ -12,5 +12,4 @@ API Documentation
predict
tfutils
utils
RL
tensorpack.tfutils package
==========================
Module contents
---------------
.. automodule:: tensorpack.tfutils
:members:
:undoc-members:
:show-inheritance:
tensorpack.tfutils.collection module
------------------------------------
......@@ -88,12 +96,3 @@ tensorpack.tfutils.varreplace module
:members:
:undoc-members:
:show-inheritance:
Module contents
---------------
.. automodule:: tensorpack.tfutils
:members:
:undoc-members:
:show-inheritance:
......@@ -15,7 +15,6 @@ import six
from six.moves import queue
import zmq
from tensorpack.models.common import disable_layer_logging
from tensorpack.callbacks import Callback
from tensorpack.tfutils.varmanip import SessionUpdate
from tensorpack.predict import OfflinePredictor
......
......@@ -13,6 +13,7 @@ import six
from six.moves import map, range
from tensorpack import *
from tensorpack.tfutils import symbolic_functions, summary, optimizer
from tensorpack.tfutils.gradproc import GlobalNormClip
from tensorpack.utils.globvars import globalns as param
......
......@@ -17,7 +17,7 @@ from ..utils.develop import building_rtfd
_LAYER_LOGGED = set()
_LAYER_REGISTERED = {}
__all__ = ['layer_register', 'disable_layer_logging', 'get_registered_layer', 'VariableHolder']
__all__ = ['layer_register', 'VariableHolder']
class VariableHolder(object):
......
......@@ -34,4 +34,4 @@ for _, module_name, _ in iter_modules(
if module_name in _TO_IMPORT:
_global_import(module_name) # import the content to tfutils.*
__all__.extend(['sessinit', 'summary', 'optimizer',
'sesscreate', 'gradproc', 'varreplace'])
'sesscreate', 'gradproc', 'varreplace', 'symbolic_functions'])
......@@ -161,7 +161,7 @@ def stack_patches(
"""
Stacked patches into grid, to produce visualizations like the following:
.. image:: https://github.com/ppwwyyxx/tensorpack/raw/master/examples/GAN/demo/CelebA-samples.jpg
.. image:: https://github.com/ppwwyyxx/tensorpack/raw/master/examples/GAN/demo/BEGAN-CelebA-samples.jpg
Args:
patch_list(list[ndarray] or ndarray): NHW or NHWC images in [0,255].
......
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