在bestmodel = load_model("BestModel.h5", custom_objects=custom_dir)行报了这个错,解决方法是在自定义的模块的__init__函数参数中加上**kwargs,如下:
class MyTransformer(layers.Layer):
def __init__(self, maxlen=64, initial_embed_dim=64, num_heads=8, ff_dim=64, **kwargs):
不是很清楚原理,但加上就不报错了