ソースを参照

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

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 用于训练不同模型,以及保存相应的路径
 # -------------------------------------------------------------------------------------------------------------------- #