Method forward

model/finetune.py:36–38

(self, x, t)

        self.batch_size = args.batch_size  # How many to process per update

    def forward(self, x, t):
        output = self.net(x)
        return output

    def observe(self, x, t, y):
        self.train()