Method predict_step

mGPT/models/base.py:35–36 · view source on GitHub ↗

(self, batch, batch_idx)

Source from the content-addressed store, hash-verified

33	return outputs
34
35	def predict_step(self, batch, batch_idx):
36	return self.forward(batch)
37
38	def on_train_epoch_end(self):
39	# Log steps and losses

nothing calls this directly

forwardMethod · 0.45

no test coverage detected