on_train_batch_start() missing 1 required positional argument: ‘dataloader_idx‘
【代码】on_train_batch_start() missing 1 required positional argument: ‘dataloader_idx‘
·
将
def on_train_batch_start(self, batch, batch_idx, dataloader_idx):
def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx):
改为
def on_train_batch_start(self, batch, batch_idx):
def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx):
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
4
0- 0
已为社区贡献3条内容
所有评论(0)