- 博客(2)
- 收藏
- 关注
转载 AI成长之路--pytorch的eval
AI成长之路--pytorch的eval 在训练完模型后,要调用模型的eval(),否则的话,有输入数据,即使不训练,它也会改变权值。(这样就导致测试结果不对) 这是model中含有batch normalization层所带来的的性质。 转载:https://blog.csdn.net/qq_38241346/article/details/106564713?ops_request_misc=%257B%2522request%255Fid%2522%253A%252216017129631919518
2020-10-03 16:21:54 92
原创 tf如何自制使用数据集
tf如何自制使用数据集 这里用tf.data.Dataset.from_tensor_slices进行加载数据集 第一步,准备numpy数据 def load_mnist(batch_size,is_training=True): if is_training: fd=open(r'data/MNIST/train-images.idx3-ubyte') loaded=np.fromfile(file=fd,dtype=np.uint8) #图片一
2020-09-29 09:27:57 413
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人