Added flag for checkpoint reading for newer versions of Pytorch#1823

ajinkya-kulkarni wants to merge 1 commit intoMegvii-BaseDetection:mainMegvii-BaseDetection/YOLOX:mainfrom

ajinkya-kulkarni:patch-1ajinkya-kulkarni/YOLOX:patch-1Copy head branch name to clipboard