pytorch lightning使用注意事项

HyperParameter

传入参数中不能有函数句柄，否则会导致报错：

TypeError: cannot pickle 'Environment' object

training_step()的参数batch

在on_train_batch_start(HOOK)中修改的batch无法直接传入training_step()的参数batch中，如果需要对DataLoader取得的数据batch进行处理，需要同样写在training_step()中

码一下以防忘记~

posted @ 2024-03-27 21:57  ilk012  阅读(19)  评论(0)  编辑  收藏  举报