From fbf644e6ccc022ea1ffb17506ef873e294c4555d Mon Sep 17 00:00:00 2001
From: Kilian Fatras <kilian.fatras.pro@gmail.com>
Date: Fri, 8 Dec 2023 17:20:35 -0500
Subject: [PATCH] remove print GPUs

---
 examples/cifar10/train_cifar10.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/cifar10/train_cifar10.py b/examples/cifar10/train_cifar10.py
index 394ace1..921aac8 100644
--- 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():