XLechter / SDT

Codes for Point Cloud Completion via Skeleton-Detail Transformer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About the pre-trained model.

RookieInAI opened this issue · comments

Hi, it seems that there are some problems about the pre-trained model and it can not load successful.

commented

hello, i also meet this issue when i used the pretrained model

Traceback (most recent call last):
File "main_benchmark.py", line 38, in
test(args)
File "/home/mnt/SDT-main/test_benchmark.py", line 39, in test
load_model(args, net, None, log_test, train=False)
File "/home/mnt/SDT-main/utils/utils.py", line 131, in load_model
ckpt = torch.load(ckpt_path)
File "/home/mnt/anaconda3/envs/sdt/lib/python3.7/site-packages/torch/serialization.py", line 587, in load
with _open_zipfile_reader(opened_file) as opened_zipfile:
File "/home/mnt/anaconda3/envs/sdt/lib/python3.7/site-packages/torch/serialization.py", line 242, in init
super(_open_zipfile_reader, self).init(torch._C.PyTorchFileReader(name_or_buffer))
RuntimeError: [enforce fail at inline_container.cc:145] . PytorchStreamReader failed reading zip archive: failed finding central directory

@RookieInAI @CRISZJ I find the model file seems incomplete as the file size should be about 180M. I updataed the pre-trained model link. Could you kindly check it?

commented

@RookieInAI @CRISZJ I find the model file seems incomplete as the file size should be about 180M. I updataed the pre-trained model link. Could you kindly check it?

it works, thanks