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