Kaiseem / PointNu-Net

PointNu-Net Project
MIT License
15 stars 1 forks source link

KeyError: 'output_stride' #5

Closed mqbin closed 1 year ago

mqbin commented 1 year ago

Hi @Kaiseem, I ran the command

python train.py --name=consep_exp --seed=888 --config=configs/consep_type_large.yaml

but I got the following error.

Traceback (most recent call last):
  File "/data/mqb/PointNu-Net/train.py", line 48, in <module>
    trainer = Trainer(config)
  File "/data/mqb/PointNu-Net/trainer.py", line 23, in __init__
    output_stride=config['model']['output_stride']
KeyError: 'output_stride'
Kaiseem commented 1 year ago

Thank you for pointing out the bug, you can specify output_stride=4 in the config file, just like pannuke.yaml. Or you can download the code again, i have corrected them.

mqbin commented 1 year ago

Thanks for your reply. I have encountered the following error. where can I download this weight file?

FileNotFoundError: [Errno 2] No such file or directory: '.torch\\models\\hrnetv2_w64_imagenet_pretrained.pth'
mqbin commented 1 year ago

Hello @Kaiseem , Sorry to bother you again. could you please provide the download link of the hrnetv2_w64_imagenet_pretrained.pth ? Thank you.

Kaiseem commented 1 year ago

Hi, sure, the checkpoints can be found here https://github.com/HRNet/HRNet-Image-Classification

here the download link is here https://1drv.ms/u/s!Aus8VCZ_C_33gQbJsUPTIj3rQu99

mqbin commented 1 year ago

Hi, sure, the checkpoints can be found here https://github.com/HRNet/HRNet-Image-Classification

here the download link is here https://1drv.ms/u/s!Aus8VCZ_C_33gQbJsUPTIj3rQu99

Thank you. It's very helpful.