fb47e1487886b33181b459dfe91b67e20a96b8f7,finetune/target_models/multiple_choice.py,MultipleChoicePipeline,feed_shape_type_def,#MultipleChoicePipeline#,41

Before Change

            "tokens": TS([self.target_dim, self.config.max_length, 2]),
            "mask": TS([self.target_dim, self.config.max_length]),
        }
        types, shapes = self._add_context_info_if_present(types, shapes)
        return (
            (types, tf.float32,),
            (shapes, TS([]),),
        )

After Change

    def feed_shape_type_def(self):
        TS = tf.TensorShape
        types = {"tokens": tf.int32, "mask": tf.float32}
        shapes = {
            "tokens": TS([self.target_dim, self.config.max_length, 2]),
            "mask": TS([self.target_dim, self.config.max_length]),
        }
        if self.config.use_auxiliary_info:
            TS = tf.TensorShape
            types["context"] = tf.float32
            shapes["context"] = TS([self.target_dim, self.config.max_length, self.config.context_dim])
        return (
            (types, tf.float32,),
            (shapes, TS([]),),
        )

Instances