小编aa *_* bb的帖子

keras 上的分类焦点损失

我想在 tf/keras 中使用分类焦点损失。Binary Focal loss 对我有用,但不是我为分类 fl 找到的代码 有人有吗?

keras tensorflow

7
推荐指数
2
解决办法
5824
查看次数

如何使用 tpu 解决 tf.keras 中的“动态维度传播失败”错误?

当我尝试使用 TPU 在 google colab 上使用我自己的图像训练图像检测器时出现此错误:

来自 /job:worker/replica:0/task:0:编译失败:要求从 hlo %convert.283 = f32[1,80,80,32]{3,2,1,0} 传播动态维度(f32[1,80,80,32]{3,2,1,0} %add.1), 元数据={op_type="FusedBatchNorm" op_name="bn_Conv1_3/FusedBatchNorm"}@{}@0 到 hlo %钳位.288 = f32[1,80,80,32]{3,2,1,0} 钳位(f32[1,80,80,32]{3,2,1,0} %broadcast.286, f32 [1,80,80,32]{3,2,1,0} %convert.283, f32[1,80,80,32]{3,2,1,0} %broadcast.287), 元数据= {op_type="Relu6" op_name="Conv1_relu_3/Relu6"},未实现。TPU 编译失败 [[节点 TPUReplicateMetadata_1(定义于:24)]]

这是代码的链接:

https://drive.google.com/open?id=1mPiod1At85RgNwHx4vYFxH38Ck16Ep1m

你知道发生了什么事吗?

一定不是图片大小的问题,也不是批量大小的问题,我已经看过了。

谢谢。

artificial-intelligence tensorflow google-colaboratory tpu

5
推荐指数
1
解决办法
819
查看次数