在保存模型文件之前,先检查路径是否存在。
This commit is contained in:
@@ -199,6 +199,13 @@ if __name__ == '__main__':
|
|||||||
if val_acc > best_val_acc:
|
if val_acc > best_val_acc:
|
||||||
best_val_acc = val_acc
|
best_val_acc = val_acc
|
||||||
best_model_path = settings.BEST_MODEL_PATH
|
best_model_path = settings.BEST_MODEL_PATH
|
||||||
|
|
||||||
|
# 检查并创建模型保存目录
|
||||||
|
model_dir = os.path.dirname(best_model_path)
|
||||||
|
if not os.path.exists(model_dir):
|
||||||
|
os.makedirs(model_dir)
|
||||||
|
print(f"创建模型保存目录: {model_dir}")
|
||||||
|
|
||||||
torch.save(model.state_dict(), settings.BEST_MODEL_PATH)
|
torch.save(model.state_dict(), settings.BEST_MODEL_PATH)
|
||||||
print(f'保存最佳模型,验证准确率: {best_val_acc:.2f}%')
|
print(f'保存最佳模型,验证准确率: {best_val_acc:.2f}%')
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user