parent
7fa54fc414
commit
6314d210c3
|
|
@ -360,7 +360,9 @@ class EnhancerDataset(TaskDataset):
|
||||||
train_data = self.train_data
|
train_data = self.train_data
|
||||||
else:
|
else:
|
||||||
train_data = worker_info.dataset.data
|
train_data = worker_info.dataset.data
|
||||||
return 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)
|
||||||
|
return len
|
||||||
|
|
||||||
def val__len__(self):
|
def val__len__(self):
|
||||||
return len(self._validation)
|
return len(self._validation)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue