Unable to plot network architecture: nnUNet_compile is enabled!无法绘制网络结构
今天跑nnunet代码,绘制网络结构报错,没有生成网络结构.pdf
在nnUNetTrainer.py代码中搜索nnUNet_compile可以找到这行报错信息:
可以看到来自_do_i_compile()函数
如果你需要绘制网络架构图,请在配置文件中禁用nnUNet_compile模式。修改代码:
注释掉其他代码,在_do_i_compile()函数中直接return False即可,之后重新训练即可生成
生成完成之后记得还原_do_i_compile()函数
@end