Skip to content

CustomMaskWarning: Custom mask layers require a config and must override get_config.

During the training of my GNN, I receive the warning

CustomMaskWarning: Custom mask layers require a config and must override get_config. When loading, the custom mask layer must be passed to the custom_objects argument.

after training finished on each file. I'm not sure where this warning comes from and whether I can ignore it. I appended my notebook, where you can find the output during training, including the warning.

Thank you in advance for your help!

binaryClassificationOneNetworkOnlyHits_2_.ipynb

To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information