gwxie/Document-Dewarping-with-Control-Points

pretrain model .pkl format can not load

AIMan-Zzx opened this issue · 2 comments

AIMan-Zzx commented

any error? do not understand

gwxie commented

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

vraivon commented

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