Hi, it seems ClearML has trouble serializing Keras model config for TF >= 2.5 when the model contains a Lambda layer with functions like K.mean. I’ve created issue # https://github.com/allegroai/clearml/issues/614 and proposed some solution that prevents a crash but still the config is not saved. I’ll make a PR for that.