pretrain model .pkl format can not load

[AIMan-Zzx]

any error? do not understand

[GuoWang]

Hi，
Try again.
If an error is reported, list more detailed information

[vraivon]

if you have only one gpu, you should modify torch.load(filePath) to torch.load(filePath, map_location='cuda:0')