Fix: correct the early return error when save_epochs=1 and optimize the clean up way by xiu-cs · Pull Request #2793 · DeepLabCut/DeepLabCut