refactor to use torch device instead of cuda boolean#595

Rusteam wants to merge 1 commit into

KaiyangZhou:masterKaiyangZhou/deep-person-reid:masterfrom

Rusteam:device-invariantRusteam/deep-person-reid:device-invariantCopy head branch name to clipboard