python代码运行中的问题,报错missing 1 required positional argument

麻烦指教一下
报错:TypeError: on_train_batch_end() missing 1 required positional argument: 'dataloader_idx'

img

train.py模块最后部分,是run的不对吗?哭了,有没有大佬可以帮忙看下,查了半天,看不明白

img

少一个参数 dataloader_idx,Ctrl p检查一下哪个函数少这个

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^