Js2Hou / Swin-CBIR

This repository contains a CBIR system that uses swin transformer to extract image's feature.
18 stars 1 forks source link

I can't run this code #1

Closed vothaianh1997 closed 1 year ago

vothaianh1997 commented 2 years ago

Traceback (most recent call last): File "generate_DB.py", line 134, in main() File "generate_DB.py", line 125, in main model = get_pretrained_model(config) File "/content/drive/MyDrive/ColabNotebooks/Swin-CBIR-main/utils.py", line 32, in get_pretrained_model checkpoint = torch.load(config.MODEL.RESUME, map_location='cpu') File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 594, in load with _open_file_like(f, 'rb') as opened_file: File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 230, in _open_file_like return _open_file(name_or_buffer, mode) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 211, in init super(_open_file, self).init(open(name, mode)) FileNotFoundError: [Errno 2] No such file or directory: 'checkpoints/swin_tiny_patch4_window7_224.pth' ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 1557) of binary: /usr/bin/python3 Traceback (most recent call last): File "/usr/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/usr/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/usr/local/lib/python3.7/dist-packages/torch/distributed/launch.py", line 193, in main() File "/usr/local/lib/python3.7/dist-packages/torch/distributed/launch.py", line 189, in main launch(args) File "/usr/local/lib/python3.7/dist-packages/torch/distributed/launch.py", line 174, in launch run(args) File "/usr/local/lib/python3.7/dist-packages/torch/distributed/run.py", line 713, in run )(*cmd_args) File "/usr/local/lib/python3.7/dist-packages/torch/distributed/launcher/api.py", line 131, in call return launch_agent(self._config, self._entrypoint, list(args)) File "/usr/local/lib/python3.7/dist-packages/torch/distributed/launcher/api.py", line 261, in launch_agent failures=result.failures, torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

generate_DB.py FAILED

Failures:

------------------------------------------------------------ Root Cause (first observed failure): [0]: time : 2022-01-16_23:32:26 host : 0da41351b232 rank : 0 (local_rank: 0) exitcode : 1 (pid: 1557) error_file: traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html ============================================================
Js2Hou commented 2 years ago

sorry, i didn't upload swin-transformer's checkpoint here, you can download it at swin_tiny_patch4_window7_224.pth, and then move it into checkpoints path.

vothaianh1997 commented 2 years ago

please.. Can you up full step how to run this code. Error in file index.txt when i upload new dataset. i'm student and i need this source.And.. Happy new Year =)

vothaianh1997 commented 2 years ago

can you give me your data .my error now .RuntimeError: Given groups=1, weight of size [96, 3, 4, 4], expected input[1, 4, 224, 224] to have 3 channels, but got 4 channels instead

Js2Hou commented 2 years ago

you can download my shared data here https://www.aliyundrive.com/s/L7PSuEpJ3Df

vothaianh1997 commented 2 years ago

can you updata in github. I can't download .aliyundrive

Js2Hou commented 2 years ago

Yes, i have done so