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):