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