Fix a bug that can not set the training device correctly#18

lgd21356 wants to merge 2 commits into

NVlabs:mainNVlabs/CALM:mainfrom

lgd21356:patch-1lgd21356/CALM:patch-1Copy head branch name to clipboard