print len
This commit is contained in:
parent
6314d210c3
commit
9b15534812
|
|
@ -361,7 +361,7 @@ class EnhancerDataset(TaskDataset):
|
||||||
else:
|
else:
|
||||||
train_data = worker_info.dataset.data
|
train_data = worker_info.dataset.data
|
||||||
len = sum([len(item) for item in train_data]) // (self.batch_size)
|
len = sum([len(item) for item in train_data]) // (self.batch_size)
|
||||||
print(len)
|
print("workers", len)
|
||||||
return len
|
return len
|
||||||
|
|
||||||
def val__len__(self):
|
def val__len__(self):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue