extract_encoder.py run fail

MaitySubhajit / SelfDocSeg

[ICDAR 2023] SelfDocSeg: A self-supervised vision-based approach towards Document Segmentation (Oral)

https://maitysubhajit.github.io/SelfDocSeg/

GNU General Public License v3.0

38 stars 2 forks source link

extract_encoder.py run fail #5

Open Cyc-striving opened 1 month ago

Cyc-striving commented 1 month ago

Traceback (most recent call last): File "/root/autodl-tmp/SelfDocSeg/pretraining/extract_encoder.py", line 633, in resnet_seq.load_state_dict(wts) # loading weights to the new encoder File "/root/miniconda3/envs/selfdocseg/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1604, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for Sequential:

MaitySubhajit commented 4 weeks ago

Please include the complete stack trace, so that we understand the issue clearly.