Post Go back to editing

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

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 ,我需要一台什么配置水平的电脑或者服务器。