8b0d454102218af317adfeeda8aefc576769e172,ludwig/models/modules/image_encoders.py,ResNetEncoder,call,#ResNetEncoder#Any#Any#Any#,203

Before Change

    def call(self, inputs, training=None, mask=None):
        hidden = tf.cast(inputs, tf.float32)

        for layer in self.layers:
            hidden = layer(hidden, training=training)

        return {"encoder_output": hidden}

After Change

        hidden = tf.cast(inputs, tf.float32)

        hidden = self.resnet(hidden, training=training)
        hidden = self.flatten(hidden, training=training)
        hidden = self.fc_stack(hidden, training=training)

        return {"encoder_output": hidden}

Instances