irfanICMLL / structure_knowledge_distillation

The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.
BSD 2-Clause "Simplified" License
695 stars 104 forks source link

I need some help from you.. #8

Closed WBS-123 closed 4 years ago

WBS-123 commented 4 years ago

from .__ext import lib as _lib, ffi as _ffi i can't find the ._ext/lib and ._ext/ffi this error arise: No module named 'libs._ext._ext' could you fix the problem for me?

irfanICMLL commented 4 years ago

Have you compiled the inplaceABN before using the code?

Compiling

Some parts of InPlace-ABN have a native CUDA implementation, which must be compiled with the following commands:

cd libs
sh build.sh
python build.py

The build.sh script assumes that the nvcc compiler is available in the current system search path. The CUDA kernels are compiled for sm_50, sm_52 and sm_61 by default. To change this (e.g. if you are using a Kepler GPU), please edit the CUDA_GENCODE variable in build.sh.

irfanICMLL commented 4 years ago

I will close this issue. You can contact me if you have any other questions.

wangq95 commented 3 years ago

@irfanICMLL Hi, I complied the InPlaceABN but got an error "ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead." when I try to run 'python build.py', the pytorch version is 1.5.0.