소스 검색

添加嵌入白盒水印训练示例

liyan 1 년 전
부모
커밋
f85f24661b
1개의 변경된 파일4개의 추가작업 그리고 0개의 파일을 삭제
  1. 4 0
      bash_run.sh

+ 4 - 0
bash_run.sh

@@ -1,3 +1,7 @@
+# 模型训练
+python run.py --model Alexnet --input_size 32 --save_path ./checkpoints/Alexnet/clean/best.pt --save_path_last ./checkpoints/Alexnet/clean/last.pt --epoch 10 --data_path ./dataset --dataset_name CIFAR-10
+# 嵌入白盒水印模型训练
+python run.py --model Alexnet --input_size 32 --save_path ./checkpoints/Alexnet/white_box_embed/best.pt --save_path_last ./checkpoints/Alexnet/white_box_embed/last.pt --epoch 10 --data_path ./dataset --dataset_name CIFAR-10 --white_box_embed True
 
 # For 用于训练不同模型,以及保存相应的路径
 # -------------------------------------------------------------------------------------------------------------------- #