mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Feature]: Change map_location to cpu

pull/623/head
YuanLiuuuuuu 2022-12-09 14:40:40 +08:00
parent 5244b66d5c
commit 976de2e518
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/model_converters/mmcls2timm.py
View File

@ -71,7 +71,7 @@ def main():
parser.add_argument('dst', help='save path') parser.add_argument('dst', help='save path')
args = parser.parse_args() args = parser.parse_args()
checkpoint = _load_checkpoint(args.src) checkpoint = _load_checkpoint(args.src, map_location='cpu')
if 'state_dict' in checkpoint: if 'state_dict' in checkpoint:
state_dict = checkpoint['state_dict'] state_dict = checkpoint['state_dict']
else: else: