Udacity DL CNN Dog_Project Note
摘要:
当显示cuda内存不够的时候,可以尝试减少batch的数量当出现“cuda runtime error (59)”时,有可能是网络的最后一层输出的数量比数据集的label少。(最后一层的输出数量多于数据集的时候不会报错,因为训练的时候由于不存在多出来的label,多出来的输出神经元就等于没训练)train的时候用model.train(), test和valid的时候用model.eval()记得... 阅读全文
posted @ 2019-02-16 16:23 bourne_lin 阅读(127) 评论(0) 推荐(0) 编辑