remove print GPUs

atong01 · Dec 8, 2023 · fbf644e · fbf644e
1 parent de6a2c3
commit fbf644e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/cifar10/train_cifar10.py b/examples/cifar10/train_cifar10.py
@@ -107,7 +107,7 @@ def train(argv):
     if FLAGS.parallel:
         net_model = torch.nn.DataParallel(net_model)
         ema_model = torch.nn.DataParallel(ema_model)
-        print(f"Training is using {torch.cuda.device_count()} GPUs!")
+
     # show model size
     model_size = 0
     for param in net_model.parameters():