From f4f09840296d53a470b390cdac1ecf22fbcec844 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Thu, 10 Jun 2021 21:41:07 +0800
Subject: [PATCH] Update mv3_large_x1_0_distill_mv3_small_x1_0.yaml

---
 .../Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml b/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
index f46d7be1..5501dd90 100644
--- a/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
+++ b/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
@@ -67,6 +67,9 @@ DataLoader:
         image_root: "./dataset/ILSVRC2012/"
         cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
         transform_ops:
+          - DecodeImage:
+              to_rgb: True
+              channel_first: False
           - RandCropImage:
               size: 224
           - RandFlipImage:
@@ -94,6 +97,9 @@ DataLoader:
         image_root: "./dataset/ILSVRC2012/"
         cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
         transform_ops:
+          - DecodeImage:
+              to_rgb: True
+              channel_first: False
           - ResizeImage:
               resize_short: 256
           - CropImage:
-- 
GitLab