Post Go back to editing

在图像训练过程中出现内存不够问题

Thread Summary

用户在使用Anaconda构建的Python 3.8.7虚拟环境中尝试运行AI8x的猫狗识别Demo时遇到内存不足的问题。官方建议检查系统配置是否满足要求,并考虑使用云计算服务。涉及产品:MAX78000。
AI Generated Content
Category: Datasheet/Specs
Product Number: MAX78000 FTHR_RevA

使用anaconda构建的python3.8.7虚拟环境,在此环境下,参考ai8x的GitHub内容进行训练,首先测试猫狗识别的Demo,在官方例程中,根据官方提示操作后,最终执行命令“python train.py --model ai85kws20net --dataset KWS_20 --confusion --evaluate --exp-load-weights-from ../ai8x-synthesis/trained/ai85-kws20-qat8-q.pth.tar -8 --device MAX78000 ”,运行后发现提示内存不够。

系统硬件配置为3060显卡、6GB显存,16GB运行内存,想了解一下该问题属于正常情况吗,如果是,问题是出在环境配置上吗,如果用 wsl2 下的 ubuntu 训练是否也面临这个问题;如果不是,如果我需要训练,不限制 batch size 的情况下,在10小时内完成该图像训练 Demo ,我需要一台什么配置水平的电脑或者服务器。