thangvubk / SoftGroup

[CVPR 2022 Oral] SoftGroup for Instance Segmentation on 3D Point Clouds
MIT License
346 stars 81 forks source link

cannot find the pretrained model:softgroup_s3dis_spconv2.pth #199

Closed mliu285 closed 7 months ago

mliu285 commented 8 months ago

Hello! Thanks for your wonderful work! When I train model from frozen backbone using S3DIS dataset, it reminds me that File Not Found of the pretrained model: ./checkpoint/softgroup_s3dis_spconv2.pth. Where can I find this model? Thank you!

2024-01-12 11:39:05,329 - INFO - Load pretrain from ./checkpoint/softgroup_s3dis_spconv2.pth Traceback (most recent call last): File "/home/featurize/work/SoftGroup/SoftGroup-main/./tools/train.py", line 206, in main() File "/home/featurize/work/SoftGroup/SoftGroup-main/./tools/train.py", line 194, in main load_checkpoint(cfg.pretrain, logger, model) File "/home/featurize/work/SoftGroup/SoftGroup-main/softgroup/util/utils.py", line 115, in load_checkpoint state_dict = torch.load(checkpoint, map_location=lambda storage, loc: storage.cuda(device)) File "/environment/miniconda3/lib/python3.10/site-packages/torch/serialization.py", line 791, in load with _open_file_like(f, 'rb') as opened_file: File "/environment/miniconda3/lib/python3.10/site-packages/torch/serialization.py", line 271, in _open_file_like return _open_file(name_or_buffer, mode) File "/environment/miniconda3/lib/python3.10/site-packages/torch/serialization.py", line 252, in init super().init(open(name, mode)) FileNotFoundError: [Errno 2] No such file or directory: './checkpoint/softgroup_s3dis_spconv2.pth'

Rurouni-z commented 8 months ago

https://github.com/thangvubk/SoftGroup?tab=readme-ov-file#pretrained-models

softgroup_s3dis_spconv2.pth