Method get_input

VDM_Decoder/diffusion_video.py:210–211 · view source on GitHub ↗

(self, batch)

Source from the content-addressed store, hash-verified

208	return loss, loss_dict
209
210	def get_input(self, batch):
211	return batch[self.input_key].to(self.dtype)
212
213	@torch.no_grad()
214	def decode_first_stage(self, z):

shared_stepMethod · 0.95

log_videoMethod · 0.95

no outgoing calls

no test coverage detected