bugfix

examples/cifar-convnet.py

@@ -154,5 +154,5 @@ if __name__ == '__main__':
 
         nr_gpu = len(args.gpu.split(','))
         trainer = QueueInputTrainer() if nr_gpu <= 1 \
-            else SyncMultiGPUTrainerParameterServer(list(range(nr_gpu)))
+            else SyncMultiGPUTrainerParameterServer(nr_gpu)
         launch_train_with_config(config, trainer)

tensorpack/train/trainers.py

@@ -36,6 +36,7 @@ def _int_to_range(x):
     if isinstance(x, int):
         assert x > 0, x
         return list(range(x))
+    return x
 
 
 class SimpleTrainer(SingleCostTrainer):