From b9d83e80b946437bb8dc0b586488fa756f52d732 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 14 Jul 2022 11:35:32 +0800 Subject: Fix a typo --- libs/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'libs') diff --git a/libs/utils.py b/libs/utils.py index 767adfc..77e6cf1 100644 --- a/libs/utils.py +++ b/libs/utils.py @@ -23,7 +23,7 @@ class BaseConfig: @dataclass class DataLoaderConfig: batch_size: int - num_worker: int + num_workers: int @dataclass class OptimConfig: @@ -148,16 +148,16 @@ class Trainer(ABC): train_loader = DataLoader(train_set, sampler=inf_sampler, batch_size=dataloader_config.batch_size, - num_workers=dataloader_config.num_worker) + num_workers=dataloader_config.num_workers) else: train_loader = DataLoader(train_set, shuffle=True, batch_size=dataloader_config.batch_size, - num_workers=dataloader_config.num_worker) + num_workers=dataloader_config.num_workers) test_loader = DataLoader(test_set, shuffle=False, batch_size=dataloader_config.batch_size, - num_workers=dataloader_config.num_worker) + num_workers=dataloader_config.num_workers) return train_loader, test_loader -- cgit v1.2.3