Open avernal1819 opened 1 year ago
there is the pre training code OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=8 run_pretrain.py --output_dir /output_dir/ --batch_size 256 --model mim_swin_base_patch4_win7 --target HOG --hog_nbins 18 --hog_bias --mask_ratio 0.75 --epochs 400 --warmup_epochs 40 --blr 1e-4 --weight_decay 0.05 --data_path /path/to/imagenet/
but I can't see where is the setup
there is the pre training code OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=8 run_pretrain.py --output_dir /output_dir/ --batch_size 256 --model mim_swin_base_patch4_win7 --target HOG --hog_nbins 18 --hog_bias --mask_ratio 0.75 --epochs 400 --warmup_epochs 40 --blr 1e-4 --weight_decay 0.05 --data_path /path/to/imagenet/
but I can't see where is the setup