move use_global_argument into ns

examples/GAN/DCGAN.py

@@ -13,7 +13,6 @@ from tensorpack.utils.viz import *
 from tensorpack.tfutils.summary import add_moving_summary
 from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope
 from tensorpack.utils.globvars import globalns as opt
-from tensorpack.utils.globvars import use_global_argument
 import tensorflow as tf
 
 from GAN import GANTrainer, RandomZData, GANModelDesc
@@ -153,7 +152,7 @@ def get_args():
     parser.add_argument('--load-size', help='size to load the original images', type=int)
     parser.add_argument('--crop-size', help='crop the original images', type=int)
     args = parser.parse_args()
-    use_global_argument(args)
+    opt.use_argument(args)
     if args.gpu:
         os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
     return args

tensorpack/utils/globvars.py

@@ -6,7 +6,7 @@
 import six
 import argparse
 
-__all__ = ['globalns', 'use_global_argument']
+__all__ = ['globalns']
 
 if six.PY2:
     class NS:
@@ -15,16 +15,18 @@ else:
     import types
     NS = types.SimpleNamespace
 
-globalns = NS()
 
+class MyNS(NS):
+    def use_argument(self, args):
+        """
+        Add the content of :class:`argparse.Namespace` to this ns.
 
-def use_global_argument(args):
-    """
-    Add the content of :class:`argparse.Namespace` to globalns.
+        Args:
+            args (argparse.Namespace): arguments
+        """
+        assert isinstance(args, argparse.Namespace), type(args)
+        for k, v in six.iteritems(vars(args)):
+            setattr(self, k, v)
 
-    Args:
-        args (argparse.Namespace): arguments
-    """
-    assert isinstance(args, argparse.Namespace), type(args)
-    for k, v in six.iteritems(vars(args)):
-        setattr(globalns, k, v)
+
+globalns = MyNS()