Esempi in C# (CSharp) per LostTech.Gradient.Samples.GPT2 Gpt2Model.Model

Esempio n. 1

        public void Train(string checkpoint, string run, int?counter, dynamic sessionConfig = null, CancellationToken cancellation = default)
        {
            Session session = sessionConfig is null
                ? Session.NewDyn(config : sessionConfig)
                : new Session();

            using (session.StartUsing()) {
                var    context = tf.placeholder(tf.int32, new TensorShape(this.batchSize, null));
                var    output  = Gpt2Model.Model(this.hParams, input: context);
                Tensor labels  = context[.., 1..];