自定义训练和测试

目录

自定义模型类

自定义训练和测试

---

自定义模型类

继承tf.keras.Model类，可以定义自己的模型类。自定义模型类要完成__init__和call函数的定义。

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel,self).__init__()
        
    def call(self,x):

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel,self).__init__()
        self.conv1 = tf.keras.layers.Conv2D(32, 3, activation='relu', name='conv1')
        self.flatten1 = tf.keras.layers.Flatten(name=