Method get_last_layer

lvdm/models/autoencoder.py:176–177 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

174	return [opt_ae, opt_disc], []
175
176	def get_last_layer(self):
177	return self.decoder.conv_out.weight
178
179	@torch.no_grad()
180	def log_images(self, batch, only_inputs=False, **kwargs):

training_stepMethod · 0.95

validation_stepMethod · 0.95

no outgoing calls

no test coverage detected