在Vitis AI 2.5的vitis-ai-pytorch 和vitis-ai-tensoflow2 conda环境中,引入了一个名为inspector的新功能,用户可以在做量化前调用inspector来检查浮点模型。Inspector根据指定的DPU目标架构,会诊断并显示神经网络的分区结果(映射到CPU还是DPU),并提示有关层未映射到DPU的原因。
The powerful deep learning system for Python now makes it easier to integrate high performance C++ code and train models on multiple machines at once PyTorch, the Python framework for quick-and-easy ...