2024年3月27日
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 阅读(19) 评论(0) 推荐(0) 编辑