python3 validate.py /home/npu/imagenet  --model volo_d1 \
  --checkpoint /root/lmm/volo-main/ --no-test-pool --apex-amp --img-size 224 -b 128