search

ucbdrive / few-shot-object-detection

Implementations of few-shot object detection benchmarks
Apache License 2.0
1.08k stars 225 forks source link

Error in running ckpt_surgery #149

Closed salehnia closed 2 years ago

salehnia commented 2 years ago

Hi everyone... when i tried to run ckpt_surgery.py like as that said here: python3 -m tools.ckpt_surgery \ --src1 checkpoints/coco/faster_rcnn/faster_rcnn_R_101_FPN_base/model_final.pth \ --method randinit \ --save-dir checkpoints/coco/faster_rcnn/faster_rcnn_R_101_FPN_all --coco i got this error: magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '__'

Can you help me for fix this error? I'm using of python 3.6 and coco format for my dataset.

Thank you