From b7e58172ce0b4dd64c233534a97ebe60040b5fe7 Mon Sep 17 00:00:00 2001 From: huypl53 Date: Wed, 11 Sep 2024 09:36:38 +0700 Subject: [PATCH] fix num worker error --- models/SSITA_adapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/SSITA_adapter.py b/models/SSITA_adapter.py index 5b30dfb..c5bbe89 100644 --- a/models/SSITA_adapter.py +++ b/models/SSITA_adapter.py @@ -70,7 +70,7 @@ def incremental_train(self, data_manager): print("The number of training dataset:", len(self.train_dataset)) self.data_manager = data_manager - self.train_loader = DataLoader(train_dataset, batch_size=self.batch_size, shuffle=True, num_workers=n8) + self.train_loader = DataLoader(train_dataset, batch_size=self.batch_size, shuffle=True, num_workers=8) test_dataset = data_manager.get_dataset(np.arange(0, self._total_classes), source="test", mode="test") self.test_loader = DataLoader(test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=8) train_dataset_for_protonet = data_manager.get_dataset(np.arange(0, self._total_classes), source="train",