工作目录就是存放训练模型、日志等文件的目录,建议选择一个空目录。
工作目录
下面是一个工作目录的例子:
训练输出文件(时间文件夹)
以时间命名的文件夹保存训练过程中的文件输出,比如日志文件:
检查点文件(checkpoints 文件夹)
checkpoints 文件夹存储训练过程中的模型,如果训练12代,则会保存12个模型文件:
不同项目如果共用 checkpoints 文件夹,旧的模型文件会被覆盖,这样可以节省硬盘空间。
最终模型文件
保存在工作目录的 DVP 文件是最终模型文件,是我们用于测试模型效果,以及部署模型的文件。体积比检查点更小,因为去掉了优化器和训练状态等数据。