MCPcopy Create free account
hub / github.com/VisionXLab/OF-Diff / get_last_layer

Method get_last_layer

ldm/models/autoencoder.py:171–172  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

169 return [opt_ae, opt_disc], []
170
171 def get_last_layer(self):
172 return self.decoder.conv_out.weight
173
174 @torch.no_grad()
175 def log_images(self, batch, only_inputs=False, log_ema=False, **kwargs):

Callers 2

training_stepMethod · 0.95
_validation_stepMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected