MCPcopy Create free account
hub / github.com/MoonInTheRiver/DiffSinger / init_train_dataloader

Method init_train_dataloader

utils/pl_utils.py:1111–1125  ·  view source on GitHub ↗
(self, model)

Source from the content-addressed store, hash-verified

1109 self.get_test_dataloaders()
1110
1111 def init_train_dataloader(self, model):
1112 self.fisrt_epoch = True
1113 self.get_train_dataloader = model.train_dataloader
1114 if isinstance(self.get_train_dataloader(), torch.utils.data.DataLoader):
1115 self.num_training_batches = len(self.get_train_dataloader())
1116 self.num_training_batches = int(self.num_training_batches)
1117 else:
1118 self.num_training_batches = float('inf')
1119 self.is_iterable_train_dataloader = True
1120 if isinstance(self.val_check_interval, int):
1121 self.val_check_batch = self.val_check_interval
1122 else:
1123 self._percent_range_check('val_check_interval')
1124 self.val_check_batch = int(self.num_training_batches * self.val_check_interval)
1125 self.val_check_batch = max(1, self.val_check_batch)
1126
1127 def init_val_dataloader(self, model):
1128 self.get_val_dataloaders = model.val_dataloader

Callers 1

get_dataloadersMethod · 0.95

Calls 1

_percent_range_checkMethod · 0.95

Tested by

no test coverage detected