From 36501e3e91ebced6bad7b52df7f14c48b762e86e Mon Sep 17 00:00:00 2001 From: karkir0003 Date: Sun, 19 Nov 2023 16:37:19 -0500 Subject: [PATCH] fixes --- dlp-cli | 2 +- training/training/core/dataset.py | 2 +- training/training/core/trainer.py | 2 -- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/dlp-cli b/dlp-cli index 1d4a27903..5b6fe147a 160000 --- a/dlp-cli +++ b/dlp-cli @@ -1 +1 @@ -Subproject commit 1d4a27903c8d9c01dd2726cd1cc1c401862fdaa7 +Subproject commit 5b6fe147ad6334b78737c6db1368045db097fb0c diff --git a/training/training/core/dataset.py b/training/training/core/dataset.py index dd06a859b..91e9ebc8b 100644 --- a/training/training/core/dataset.py +++ b/training/training/core/dataset.py @@ -109,7 +109,7 @@ class DefaultImageDatasets(Enum): MNIST = "MNIST" FASHION_MNIST = "FashionMNIST" KMNIST = "KMNIST" - CIFAR = "CIFAR10" + CIFAR10 = "CIFAR10" class ImageDefaultDatasetCreator(TrainTestDatasetCreator): diff --git a/training/training/core/trainer.py b/training/training/core/trainer.py index a25c31bc9..45fe27539 100644 --- a/training/training/core/trainer.py +++ b/training/training/core/trainer.py @@ -204,8 +204,6 @@ def _train_init(self): super()._train_init() def _train_step(self, inputs: torch.Tensor, labels: torch.Tensor): - print(f"input shape: {inputs.shape}") - print(f"labels shape: {labels.shape}") super()._train_step(inputs, labels) self._train_correct += self.compute_correct(self._outputs, labels)