parent
1d848f409a
commit
74366f733c
@ -1,3 +0,0 @@
|
||||
#!/bin/bash
|
||||
NUM_PROC=2
|
||||
CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@" --model efficientnet_b0 --weight-decay 2.5e-2 --drop 0.2 --drop-path 0.2 --lr 0.002 --batch-size 192 --epochs 400 --sched cosine --opt adabelief --workers 8 --warmup-lr 1e-4 --aa rand-m9-mstd0.5 --remode pixel --reprob 0.2 --amp --bn-momentum 0.1 --mixup 0.2 --mixup-off-epoch 400 --min-lr 1e-5
|
Loading…
Reference in new issue