From 90d5f41c10f267bc65d25139fd250f8353aa98f9 Mon Sep 17 00:00:00 2001 From: towzeur Date: Tue, 16 Mar 2021 23:19:22 +0100 Subject: [PATCH] Update simclr.py --- simclr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/simclr.py b/simclr.py index e022dca6..36191113 100644 --- a/simclr.py +++ b/simclr.py @@ -63,7 +63,7 @@ def train(self, train_loader): n_iter = 0 logging.info(f"Start SimCLR training for {self.args.epochs} epochs.") - logging.info(f"Training with gpu: {self.args.disable_cuda}.") + logging.info(f"Training with gpu: {not self.args.disable_cuda}.") for epoch_counter in range(self.args.epochs): for images, _ in tqdm(train_loader):