03 2024 档案

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 阅读(42) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示