Jupyter 内核死于尝试使用自定义层拟合 Keras 模型
Jupyter kernel dies trying to fit a Keras model with a custom layer
在尝试拟合 Keras 模型时,笔记本在本地和云端 (kaggle) 死机(忘记任何导入或变量,停止单元的执行)。仅当体系结构中存在自定义层 SingularityExtractor2D
时才会出现此问题。
您可以在这里找到笔记本:
Github
自定义回调 GateOfLearning
已经过多次测试,每次都能与任何模型架构完美配合。
笔记本已经运行GPU和CPU,问题依旧
当调用 tf.ones
函数时,形状必须是 (spatial_0, ..., spatialN-1, in_channels, out_channels)
而不是调用时层获得的张量的形状 (batch_size, spatial_0, ..., spatial_N-1, channels)
在尝试拟合 Keras 模型时,笔记本在本地和云端 (kaggle) 死机(忘记任何导入或变量,停止单元的执行)。仅当体系结构中存在自定义层 SingularityExtractor2D
时才会出现此问题。
您可以在这里找到笔记本: Github
自定义回调 GateOfLearning
已经过多次测试,每次都能与任何模型架构完美配合。
笔记本已经运行GPU和CPU,问题依旧
当调用 tf.ones
函数时,形状必须是 (spatial_0, ..., spatialN-1, in_channels, out_channels)
而不是调用时层获得的张量的形状 (batch_size, spatial_0, ..., spatial_N-1, channels)