Jupyter kernel 死于试图用自定义层安装 Keras model

Question

在本地和云 (kaggle) 上，当尝试适应 Keras model 时，笔记本死机（忘记任何导入或变量，停止单元的执行）。仅当架构中存在自定义层SingularityExtractor2D时才会出现此问题.

你可以在这里找到笔记本： Github

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

Answer 1

当调用tf.ones function 时，形状必须是(spatial_0, ..., spatialN-1, in_channels, out_channels)而不是调用时层获得的张量的形状(batch_size, spatial_0, ..., spatial_N-1, channels)