修改前:
def configure_optimizers(self):
self.train_len = len(self.train_dataloader())
...
修改后
def configure_optimizers(self):
self.train_len = len(self.trainer._data_connector._train_dataloader_source.dataloader())
...
修改前:
def configure_optimizers(self):
self.train_len = len(self.train_dataloader())
...
修改后
def configure_optimizers(self):
self.train_len = len(self.trainer._data_connector._train_dataloader_source.dataloader())
...