docker run --shm-size=8g --privileged --network=host --rm -it -w $PWD -v $PWD:$PWD -v /dataset:/dataset -v /model_zoo:/model_zoo \
    -v $HOME:$HOME \
    oneflow-test:0.2 \
    bash