cifar10模型训练完,用于识别单个图片报错解决方案

cifar10模型训练完,用于识别单个图片报错解决方案_第1张图片

(InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [72,384] rhs shape= [9216,384]
[[Node: save/Assign_5 = Assign[T=DT_FLOAT, _class=["loc:@local3/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](local3/weights, save/RestoreV2_5)]])

解决这种问题一般步骤

1.chenkpoint 先清除再运行

2.设置图片的格式与训练图片格式要一致


你可能感兴趣的:(cifar10模型训练完,用于识别单个图片报错解决方案)