pytorch lightning使用注意事项

HyperParameter 传入参数中不能有函数句柄，否则会导致报错： TypeError: cannot pickle 'Environment' object training_step()的参数batch 在on_train_batch_start(HOOK)中修改的batch无法直接传入t
posted @ 2024-03-27 21:57  ilk012  阅读(22)  评论(0)  编辑  收藏  举报