添加分支说明

README.md

@@ -1,6 +1,12 @@
 ## pytorch图片分类训练框架
 >代码兼容性较强，使用的是一些基本的库、基础的函数  
 >在argparse中可以选择使用wandb，能在wandb网站中生成可视化的训练过程
+
+## 分支说明
+- `master`分支，废弃
+- `demo`分支，废弃
+- `whitebox_wm_integrate`分支，包含图像分类模型定义、模型训练、模型验证、模型水印训练、模型水印验证
+
 ### 1，环境
 >torch：https://pytorch.org/get-started/previous-versions/
 >```