Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training script inquiry #106

Open
zw615 opened this issue Nov 27, 2022 · 0 comments
Open

Training script inquiry #106

zw615 opened this issue Nov 27, 2022 · 0 comments

Comments

@zw615
Copy link

zw615 commented Nov 27, 2022

Hi Xiaohan, thanks for the great work!

I wonder if you could provide the original training script to reproduce the results of Table 5 (200 epochs with Autoaugment, label smoothing and mixup) in the RepVGG paper. It seems now the script in the README only includes the one for Table 4 (with epoch set to 300):

python -m torch.distributed.launch --nproc_per_node 8 --master_port 12349 main.py --arch [model name] --data-path [/path/to/imagenet] --batch-size 32 --tag train_from_scratch --output-dir /path/to/save/the/log/and/checkpoints --opts TRAIN.EPOCHS 300 TRAIN.BASE_LR 0.1 TRAIN.WEIGHT_DECAY 1e-4 TRAIN.WARMUP_EPOCHS 5 MODEL.LABEL_SMOOTHING 0.1 AUG.PRESET weak AUG.MIXUP 0.0 DATA.DATASET imagenet DATA.IMG_SIZE 224

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant