Poolformer s24

cv/classification/poolformer_s24/README.md

@@ -0,0 +1,65 @@
+## Poolformer-S24
+
+### Installation
+- Install the latest version of OneFlow
+```bash
+python3 -m pip install oneflow -f https://staging.oneflow.info/branch/master/[PLATFORM]
+```
+Find more information on [install oneflow](https://github.com/Oneflow-Inc/oneflow#install-oneflow)
+
+- Install flowvision
+
+Then install the latest stable release of flowvision
+
+```bash
+pip install flowvision==0.2.1
+```
+
+- Install other requirements
+```bash
+python3 -m pip install -r requirements.txt
+```
+
+
+### Dataset
+#### ImageNet
+For ImageNet dataset, you can download it from http://image-net.org/.
+
+### Code Structure
+
+
+  ```
+    .
+    ├── configs -> ../configs
+    │   ├── ...
+    │   └── default_settings.yaml
+    ├── data -> ../data
+    │   ├── __init__.py
+    │   ├── build.py
+    │   ├── cached_image_folder.py
+    │   ├── samplers.py
+    │   └── zipreader.py
+    ├── utils.py -> ../utils.py
+    ├── config.py -> ../config.py
+    ├── logger.py -> ../logger.py
+    ├── lr_scheduler.py -> ../lr_scheduler.py
+    ├── optimizer.py -> ../optimizer.py
+    ├── main.py
+    ├── train.sh
+    └── infer.sh
+  ```
+
+
+
+### Training
+You can use bash script `train.sh` to train this model.
+```````
+sh train.sh
+```````
+
+### Inference
+
+Bash script `infer.sh` is used to infer the trained model.
+```````
+sh infer.sh
+```````

cv/classification/poolformer_s24/config.py

@@ -0,0 +1 @@
+../config.py

cv/classification/poolformer_s24/configs

@@ -0,0 +1 @@
+../configs/

cv/classification/poolformer_s24/data

@@ -0,0 +1 @@
+../data/

cv/classification/poolformer_s24/infer.sh

@@ -0,0 +1,15 @@
+export PYTHONPATH=$PWD:$PYTHONPATH
+set -aux
+
+GPU_NUMS=1
+PORT=12346
+MODEL_ARCH="poolformer_s24"
+
+python3 -m oneflow.distributed.launch \
+        --nproc_per_node $GPU_NUMS \
+        --master_addr 127.0.0.1 \
+        --master_port $PORT \
+        main.py \
+        --cfg configs/default_settings.yaml \
+        --model_arch $MODEL_ARCH \
+        --throughput

cv/classification/poolformer_s24/logger.py

@@ -0,0 +1 @@
+../logger.py

cv/classification/poolformer_s24/lr_scheduler.py

@@ -0,0 +1 @@
+../lr_scheduler.py

cv/classification/poolformer_s24/main.py

@@ -0,0 +1 @@
+../main.py

cv/classification/poolformer_s24/optimizer.py

@@ -0,0 +1 @@
+../optimizer.py

cv/classification/poolformer_s24/requirements.txt

@@ -0,0 +1 @@
+../requirements.txt

cv/classification/poolformer_s24/train.sh

@@ -0,0 +1,19 @@
+export PYTHONPATH=$PWD:$PYTHONPATH
+set -aux
+
+GPU_NUMS=4
+PORT=12346
+MODEL_ARCH="poolformer_s24"
+BATCH_SIZE=64
+LEARNING_RATE=1e-4
+
+python3 -m oneflow.distributed.launch \
+        --nproc_per_node $GPU_NUMS \
+        --master_addr 127.0.0.1 \
+        --master_port $PORT \
+        main.py \
+        --cfg configs/default_settings.yaml \
+        --model_arch $MODEL_ARCH \
+        --batch-size $BATCH_SIZE \
+        --lr $LEARNING_RATE  
+

cv/classification/poolformer_s24/utils.py

@@ -0,0 +1 @@
+../utils.py