按照这个指南https://keras.io/guides/customizing_what_happens_in_fit/,我已经创建了一个自定义版本的train_step,在调用model.fit@tf.function ''' Adapted from https://keras.io/guides/customizin
我正在尝试使用子类化API's来做几乎所有的事情。我已经创建了我的custom转换块,如下所示: def __init__(self, in_features, kernel_size当我尝试使用自定义的.compile()、.train()和.evaluate()调用.compile()、.train()和.evaluate()时,问题就来了。下面的代码显示了我是如何创建它们的:
class M