lyhue1991/torchkeras

torchkeras怎么支持每隔n个step保存一次模型

terminator123 opened this issue · 1 comments

terminator123 commented

大模型微调一个epoch太久了，想看下中间按step的loss和ckpt，博主知道怎么设置吗

lyhue1991 commented

可以设置 dl_train.size=300，会视作300个step为一个epoch，注意设置dataloader的shuffle=True