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'
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'