aribornstein · February 14, 2021 15:14
diff --git a/lightning_forward_vs_training_step.py b/lightning_forward_vs_training_step.py
 def forward(...):
    embeddings = self.encoder(x)

 def training_step(...):
    x, y = ...
    z = self.encoder(x)
    pred = self.decoder(z)
    ...
	def forward(...):
	embeddings = self.encoder(x)

	def training_step(...):
	x, y = ...
	z = self.encoder(x)
	pred = self.decoder(z)
	...