Lucycoding的博客

原创 nnunet 报错cannot import name ‘GradScaler‘ from ‘torch‘

这篇文章主要介绍了在nnUNet框架中进行医学图像分割训练时遇到的两个关键问题及其解决方案： Windows系统下的多线程支持问题：由于Windows系统对多线程支持有限，需要在初始化时禁用数据增强的多线程处理，强制使用单线程数据加载器。 自动混合精度(AMP)训练问题：原代码中的GradScaler导入路径错误，修改为正确路径后仍存在问题，因此完全禁用了AMP功能，改用FP32精度进行训练。 文章详细展示了修改后的nnUNetTrainer.py代码实现，包括禁用多线程的环境变量设置、移除AMP相关代码、