From c8c9abf0a0d57ca5f1dcda23fc02136e2c04aae2 Mon Sep 17 00:00:00 2001
From: procr <chenrong06@baidu.com>
Date: Fri, 15 Jan 2021 16:01:01 +0800
Subject: [PATCH] support VGG16/19 for kunlun (#560)

---
 .../quick_start/VGG16_finetune_kunlun.yaml    | 72 +++++++++++++++++++
 .../quick_start/VGG19_finetune_kunlun.yaml    | 72 +++++++++++++++++++
 docs/zh_CN/extension/train_on_xpu.md          |  7 ++
 3 files changed, 151 insertions(+)
 create mode 100644 configs/quick_start/VGG16_finetune_kunlun.yaml
 create mode 100644 configs/quick_start/VGG19_finetune_kunlun.yaml

diff --git a/configs/quick_start/VGG16_finetune_kunlun.yaml b/configs/quick_start/VGG16_finetune_kunlun.yaml
new file mode 100644
index 00000000..bb358cf7
--- /dev/null
+++ b/configs/quick_start/VGG16_finetune_kunlun.yaml
@@ -0,0 +1,72 @@
+mode: 'train'
+ARCHITECTURE:
+    name: 'VGG16'
+    params:
+        stop_grad_layers: 5
+pretrained_model: "./pretrained/VGG16_pretrained"
+model_save_dir: "./output/"
+classes_num: 102
+total_images: 1020
+save_interval: 1
+validate: True
+valid_interval: 1
+epochs: 20
+topk: 5
+image_shape: [3, 224, 224]
+
+LEARNING_RATE:
+    function: 'Cosine'
+    params:
+        lr: 0.0005
+
+OPTIMIZER:
+    function: 'Momentum'
+    params:
+        momentum: 0.9
+    regularizer:
+        function: 'L2'
+        factor: 0.00001
+
+TRAIN:
+    batch_size: 20
+    num_workers: 0
+    file_list: "./dataset/flowers102/train_list.txt"
+    data_dir: "./dataset/flowers102/"
+    shuffle_seed: 0
+    transforms:
+        - DecodeImage:
+            to_rgb: True
+            to_np: False
+            channel_first: False
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1./255.
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - ToCHWImage:
+
+VALID:
+    batch_size: 20
+    num_workers: 0
+    file_list: "./dataset/flowers102/val_list.txt"
+    data_dir: "./dataset/flowers102/"
+    shuffle_seed: 0
+    transforms:
+        - DecodeImage:
+            to_rgb: True
+            to_np: False
+            channel_first: False
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - ToCHWImage:
diff --git a/configs/quick_start/VGG19_finetune_kunlun.yaml b/configs/quick_start/VGG19_finetune_kunlun.yaml
new file mode 100644
index 00000000..7b32e462
--- /dev/null
+++ b/configs/quick_start/VGG19_finetune_kunlun.yaml
@@ -0,0 +1,72 @@
+mode: 'train'
+ARCHITECTURE:
+    name: 'VGG19'
+    params:
+        stop_grad_layers: 5
+pretrained_model: "./pretrained/VGG19_pretrained"
+model_save_dir: "./output/"
+classes_num: 102
+total_images: 1020
+save_interval: 1
+validate: True
+valid_interval: 1
+epochs: 20
+topk: 5
+image_shape: [3, 224, 224]
+
+LEARNING_RATE:
+    function: 'Cosine'
+    params:
+        lr: 0.0005
+
+OPTIMIZER:
+    function: 'Momentum'
+    params:
+        momentum: 0.9
+    regularizer:
+        function: 'L2'
+        factor: 0.00001
+
+TRAIN:
+    batch_size: 20
+    num_workers: 0
+    file_list: "./dataset/flowers102/train_list.txt"
+    data_dir: "./dataset/flowers102/"
+    shuffle_seed: 0
+    transforms:
+        - DecodeImage:
+            to_rgb: True
+            to_np: False
+            channel_first: False
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1./255.
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - ToCHWImage:
+
+VALID:
+    batch_size: 20
+    num_workers: 0
+    file_list: "./dataset/flowers102/val_list.txt"
+    data_dir: "./dataset/flowers102/"
+    shuffle_seed: 0
+    transforms:
+        - DecodeImage:
+            to_rgb: True
+            to_np: False
+            channel_first: False
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - ToCHWImage:
diff --git a/docs/zh_CN/extension/train_on_xpu.md b/docs/zh_CN/extension/train_on_xpu.md
index 7f7d50f1..8f984683 100644
--- a/docs/zh_CN/extension/train_on_xpu.md
+++ b/docs/zh_CN/extension/train_on_xpu.md
@@ -23,3 +23,10 @@
 * 命令：
 
 ```python3.7 tools/static/train.py -c configs/quick_start/HRNet_W18_C_finetune.yaml -o is_distributed=False -o use_cpu=False  -o use_xpu=True -o use_gpu=False```
+
+
+### VGG16/19
+* 命令：
+
+```python3.7 tools/static/train.py -c configs/quick_start/VGG16_finetune_kunlun.yaml -o use_gpu=False -o use_cpu=False -o use_xpu=True -o is_distributed=False```
+```python3.7 tools/static/train.py -c configs/quick_start/VGG19_finetune_kunlun.yaml -o use_gpu=False -o use_cpu=False -o use_xpu=True -o is_distributed=False```
-- 
GitLab