heroacool的专栏

用nvidia-smi查看显存看pid 5466是什么线程ps aux | grep 5466原来是chrome，退出chrome即可正常运行pytorch

detectron2训练自有数据做检测

背景：升级了pytorch到1.9，detectron2没有pytorch1.9对应的预编译好的包，所以需要从源码编译。系统环境：ubuntu16.04步骤安装torchconda create -n conda37 python=3.7conda activate conda37pip install torch torchvision torchaudio编译detectron2git clone https://github.com/facebookresearch/de

系统环境：ubuntu16.04cuda11torch包的版本torch 1.8.1torchaudio 0.8.1torchvision 0.9.1运行官方的tv-training-code.py时出现错误RuntimeError: DataLoader worker (pid 18906) is killed by signal: Segmentation fault.找了半天应该是dat

一遍一遍的看，问题数量逐渐收敛。model.roi_heads是什么？