(self, model)
| 1094 | raise TypeError("Gradient accumulation supports only int and dict types") |
| 1095 | |
| 1096 | def get_dataloaders(self, model): |
| 1097 | if not self.testing: |
| 1098 | self.init_train_dataloader(model) |
| 1099 | self.init_val_dataloader(model) |
| 1100 | else: |
| 1101 | self.init_test_dataloader(model) |
| 1102 | |
| 1103 | if self.use_ddp: |
| 1104 | dist.barrier() |
| 1105 | if not self.testing: |
| 1106 | self.get_train_dataloader() |
| 1107 | self.get_val_dataloaders() |
| 1108 | else: |
| 1109 | self.get_test_dataloaders() |
| 1110 | |
| 1111 | def init_train_dataloader(self, model): |
| 1112 | self.fisrt_epoch = True |
no test coverage detected