fix lint

tensorpack/graph_builder/training.py

@@ -143,7 +143,7 @@ class SyncMultiGPUParameterServerBuilder(DataParallelBuilder):
         # self.train_op = tf.group(*ops)
         # return
 
-        grads = average_grads(grad_list, colocate=True)
+        grads = average_grads(grad_list, colocation=True)
         # grads = grad_list[0]
 
         opt = get_opt_fn()

tensorpack/graph_builder/utils.py

@@ -172,7 +172,7 @@ class OverrideCachingDevice(object):
         if size is None:
             # print(args, kwargs)
             return getter(*args, **kwargs)
-        if kwargs.get('trainable', True) == False:
+        if not kwargs.get('trainable', True):
             return getter(*args, **kwargs)
         if size < self.small_variable_size_threshold:
             device_name = self.device_for_small_variables