diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 11:35:32 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 11:35:32 +0800 |
| commit | b9d83e80b946437bb8dc0b586488fa756f52d732 (patch) | |
| tree | 933a0067fa2e8803e6ffa9f7899e8d781b4905b3 | |
| parent | d547f7384afc61661d8df09763c8102a916143ed (diff) | |
Fix a typo
| -rw-r--r-- | libs/utils.py | 8 |
1 files changed, 4 insertions, 4 deletions
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 |