pytorch lightning使用注意事项
HyperParameter
传入参数中不能有函数句柄,否则会导致报错:
TypeError: cannot pickle 'Environment' object
training_step()的参数batch
在on_train_batch_start(HOOK)中修改的batch无法直接传入training_step()的参数batch中,如果需要对DataLoader取得的数据batch进行处理,需要同样写在training_step()中
码一下以防忘记~
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET10 - 预览版1新功能体验(一)