Jupyter 内核在尝试使用自定义层拟合 Keras 模型时死机答案

【问题标题】：Jupyter kernel dies trying to fit a Keras model with a custom layerJupyter 内核在尝试使用自定义层拟合 Keras 模型时死机
【发布时间】：2021-08-20 15:41:31
【问题描述】：

在尝试拟合 Keras 模型时，无论是在本地还是在云 (kaggle) 上，笔记本都会死掉（忘记任何导入或变量，停止单元的执行）。仅当架构中存在自定义层 SingularityExtractor2D 时才会出现此问题。

您可以在此处找到笔记本： Github

自定义回调GateOfLearning 已经过多次测试，每次都能与任何模型架构完美配合。笔记本已经在 GPU 和 CPU 上运行，问题依然存在。

【问题讨论】：

您可以检查this & this SO 线程是否对您有帮助。

标签： python tensorflow keras neural-network jupyter-notebook

【解决方案1】：

调用tf.ones函数时，形状必须是(spatial_0, ..., spatialN-1, in_channels, out_channels)，而不是图层在收到调用(batch_size, spatial_0, ..., spatial_N-1, channels)时得到的张量形状

【讨论】：