readme.md

origins

based on https://github.com/microsoft/Swin-Transformer#20240103

main.py and utils/utils_ema.py is modified from https://github.com/microsoft/Swin-Transformer#20240103, based on https://github.com/facebookresearch/ConvNeXt#20240103

Run

To run training for vrwkv6 or vrwkv7 model, run:

mkdir tmp
# Downloading your ImageNet dataset
# Ensuring it is in the directory named "imagenet"
bash train-vrwkv6.sh
# OR
bash train-vrwkv7.sh