From 98e1c22bd8a64a73a75e09ccd36d21ea21329117 Mon Sep 17 00:00:00 2001
From: root <root@heros-common-service-application-foundation-c866d95c8-dx4gz>
Date: Tue, 24 Sep 2024 14:28:33 +0800
Subject: [PATCH]

---
 configuration.json | 300 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 300 insertions(+)
 create mode 100644 configuration.json

diff --git a/configuration.json b/configuration.json
new file mode 100644
index 0000000..f96f9e6
--- /dev/null
+++ b/configuration.json
@@ -0,0 +1,300 @@
+{
+    "framework":"pytorch",
+    "task":"image-classification",
+
+    "pipeline":{
+        "type":"nextvit-small_image-classification_Dailylife-labels"
+    },
+
+    "model":{
+        "type": "ClassificationModel",
+        "mm_model": {
+            "type": "ImageClassifier",
+            "pretrained": null,
+            "backbone": {
+                "type": "NextViT",
+                "arch": "small",
+                "path_dropout": 0.2
+            },
+            "neck": {
+                "type": "GlobalAveragePooling"
+            },
+            "head": {
+                "type": "LinearClsHead",
+                "num_classes": 1296,
+                "in_channels": 1024,
+                "loss": {
+                    "type": "CrossEntropyLoss",
+                    "loss_weight": 1.0
+                },
+                "init_cfg": [{
+                    "type": "TruncNormal",
+                    "layer": "Linear",
+                    "std": 0.02,
+                    "bias": 0.0
+                }]
+            }
+        }
+    },
+
+    "dataset": {
+        "classes": null
+    },
+
+    "preprocessor": {
+        "type": "image-classification-bypass-preprocessor",
+        "train": [
+            {   
+                "type": "LoadImageFromFile"
+            },
+            {
+                "type": "RandomResizedCrop",
+                "size": 224,
+                "backend": "pillow",
+                "interpolation": "bicubic"
+            },
+            {
+                "type": "RandomFlip",
+                "flip_prob": 0.5,
+                "direction": "horizontal"
+            },
+            {
+                "type": "RandAugment",
+                "policies": [
+                    {
+                        "type": "AutoContrast"
+                    },
+                    {
+                        "type": "Equalize"
+                    },
+                    {
+                        "type": "Invert"
+                    },
+                    {
+                        "type": "Rotate",
+                        "magnitude_key": "angle",
+                        "magnitude_range": [0, 30]
+
+                    },
+                    {
+                        "type": "Posterize",
+                        "magnitude_key": "bits",
+                        "magnitude_range": [4, 0]
+
+                    },
+                    {
+                        "type": "Solarize",
+                        "magnitude_key": "thr",
+                        "magnitude_range": [256, 0]
+
+                    },
+                    {
+                        "type": "SolarizeAdd",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 110]
+
+                    },
+                    {
+                        "type": "ColorTransform",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.9]
+
+                    },
+                    {
+                        "type": "Contrast",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.9]
+
+                    },
+                    {
+                        "type": "Brightness",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.9]
+
+                    },
+                    {
+                        "type": "Sharpness",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.9]
+
+                    },
+                    {
+                        "type": "Shear",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.3],
+                        "direction": "horizontal"
+
+                    },
+                    {
+                        "type": "Shear",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.3],
+                        "direction": "vertical"
+
+                    },
+                    {
+                        "type": "Translate",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.45],
+                        "direction": "horizontal"
+
+                    },
+                    {
+                        "type": "Translate",
+                        "magnitude_key": "magnitude",
+                        "magnitude_range": [0, 0.45],
+                        "direction": "vertical"
+
+                    }
+                ],
+                "num_policies": 2,
+                "total_level": 10,
+                "magnitude_level": 9,
+                "magnitude_std": 0.5,
+                "hparams": {
+                    "pad_val": [104, 116, 124],
+                    "interpolation": "bicubic"
+                }
+            },
+            {
+                "type": "RandomErasing",
+                "erase_prob": 0.25,
+                "mode": "rand",
+                "min_area_ratio": 0.02,
+                "max_area_ratio": 0.3333333333333333,
+                "fill_color": [103.53, 116.28, 123.675],
+                "fill_std": [57.375, 57.12, 58.395]
+            },
+            {
+                "type": "Normalize",
+                "mean": [123.675, 116.28, 103.53],
+                "std": [58.395, 57.12, 57.375],
+                "to_rgb": true
+            },
+            {
+                "type": "ImageToTensor",
+                "keys": ["img"]
+            },
+            {
+                "type": "ToTensor",
+                "keys": ["gt_label"]
+            },
+            {
+                "type": "Collect",
+                "keys": ["img", "gt_label"]
+            }
+        ],
+        
+        "val": [
+            {   
+                "type": "LoadImageFromFile"
+            },
+            {
+                "type": "Resize",
+                "size": [256, -1],
+                "backend": "pillow",
+                "interpolation": "bicubic"
+            },
+            {
+                "type": "CenterCrop",
+                "crop_size": 224
+            },
+            {
+                "type": "Normalize",
+                "mean": [123.675, 116.28, 103.53],
+                "std": [58.395, 57.12, 57.375],
+                "to_rgb": true
+            },
+            {
+                "type": "ImageToTensor",
+                "keys": ["img"]
+            },
+            {
+                "type": "Collect",
+                "keys": ["img"]
+            }
+        ]
+    },
+
+    "train": {
+        "dataloader": {
+            "batch_size_per_gpu": 32,
+            "workers_per_gpu": 4
+        },
+        "max_epochs": 1,
+        "runner": {
+            "type": "EpochBasedRunner",
+            "max_epochs": 300
+        },
+        "evaluation": {
+            "interval": 1,
+            "metric": "accuracy",
+            "save_best": "auto"
+        },
+        "checkpoint_config": {
+            "interval": 1,
+            "max_keep_ckpts": 20,
+            "create_symlink": true
+        },
+        "log_config": {
+            "interval": 100,
+            "hooks": [
+                {
+                    "type": "TextLoggerHook"
+                }
+            ]
+        },
+        "custom_hooks": [
+            {
+                "type": "EMAHook",
+                "momentum": 4e-05,
+                "priority": "ABOVE_NORMAL"
+            }
+        ],
+        "workflow": [
+            ["train", 1]
+        ],
+        "work_dir": "./work_dir",
+        "optimizer": {
+            "type": "AdamW",
+            "lr": 0.001,
+            "weight_decay": 0.1,
+            "eps": 1e-08,
+            "betas": [0.9, 0.999],
+            "paramwise_cfg": {
+                "norm_decay_mult": 0.0,
+                "bias_decay_mult": 0.0,
+                "custom_keys": {
+                    ".cls_token": {
+                        "decay_mult": 0.0
+                    },
+                    ".pos_embed": {
+                        "decay_mult": 0.0
+                    }
+                }
+            }
+        },
+        "optimizer_config": {
+            "grad_clip": {
+                "max_norm": 5.0
+            }
+        },
+        "lr_config": {
+            "policy": "CosineAnnealing",
+            "by_epoch": false,
+            "min_lr_ratio": 0.01,
+            "warmup": "linear",
+            "warmup_ratio": 0.001,
+            "warmup_iters": 20,
+            "warmup_by_epoch": true
+        }
+    },
+
+    "evaluation": {
+        "dataloader": {
+            "batch_size_per_gpu": 32,
+            "workers_per_gpu": 4
+        },
+        "metrics": ["accuracy"]
+    }
+}