使用sipeed提供的maix_train进行k210本地模型训练的时候,出现了以下问题
查找之前的log信息会发现以下错误
同时,
经过查找发现这个报错是由于显存不足引起,可以直接在文件中加入如下代码,让tensorflow自动进行显存分配
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
于是寻找到出现报错的train_function
这个函数在Train类中的classify和detect部分都有各自的实现,找到__init__
两个文件搜索train不难找到
train/classifier/__init__.py
和train/detector/__init__.py
中的两个train()函数,在import语句后加入上述代码,再次进行训练,问题解决