HXLH50K

Error of a layer with multi input

2

I want to draw a heatmap for "Attention UNet", which has attention gate need two input. When code run the loop in class FeatureExtractor: ``` for name, module in self.model._modules.items():...

我的输入shape中有None，所以这一行会报错。用[https://github.com/titu1994/Keras-Group-Normalization/blob/master/group_norm.py](https://github.com/titu1994/Keras-Group-Normalization/blob/master/group_norm.py)可以运行。