自定义训练和测试

目录

自定义模型类

自定义训练和测试

---

自定义模型类

继承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='flatten1')
        self.dense1 = tf.keras.layers.Dense(units=128, activation='relu', name='dense1')
        self.dense2 = tf.keras.layers.Dense(units=10, name='logits')
    def call(self,x):
        x = self.conv1(x)
        x = self.flatten1(x)
        x = self.dense1(x)
        x = self.dense2(x)
        return x

 通过model