如何将.pt文件转换为.pth文件？

Question

我正在尝试将 .pt 文件转换为 .pth 文件。我想保存模型的整个状态，但我不知道如何修改 train.py。

我使用的数据集来自这个GitHub page/repo - https://github.com/ultralytics/yolov5

这是 python 文件的链接 （谷歌驱动器）-train.py：https://drive.google.com/file/d/1mywUBDaME6AfU7JNTdpd3fWARl0F6F8f/view?usp=sharing

提前致谢。