PingoLH / FCHarDNet

Fully Convolutional HarDNet for Segmentation in Pytorch
MIT License
195 stars 52 forks source link

Error on test #20

Open sctrueew opened 4 years ago

sctrueew commented 4 years ago

Hello,

When I use this command to test I got error:

python validate.py --config configs\hardnet.yml --model_path weights\hardnet70_cityscapes_model.pkl

Error: File "validate.py", line 12, in from pytorch_bn_fusion.bn_fusion import fuse_bn_recursively ModuleNotFoundError: No module named 'pytorch_bn_fusion.bn_fusion'

torch verison: 1.1.0,

PingoLH commented 4 years ago

Hi, please clone the repo with --recursive. You can also clone https://github.com/MIPT-Oulu/pytorch_bn_fusion.git under FCHarDNet/