updated typos to v1.16.15 and fix typos

This commit is contained in:
Yuta Hayashibe
2023-10-01 21:51:24 +09:00
parent c8fcfd4581
commit 27f9b6ffeb
19 changed files with 63 additions and 45 deletions

View File

@@ -4658,7 +4658,7 @@ class ImageLoadingDataset(torch.utils.data.Dataset):
# collate_fn用 epoch,stepはmultiprocessing.Value
class collater_class:
class collator_class:
def __init__(self, epoch, step, dataset):
self.current_epoch = epoch
self.current_step = step